ACM题目————数独

描述

数独是一种运用纸、笔进行演算的逻辑游戏。玩家需要根据9×9盘面上的已知数字，推理出所有剩余空格的数字，并满足每一行、每一列、每一个3*3宫内的 数字均含1-9，不重复。 每一道合格的数独谜题都有且仅有唯一答案，推理方法也以此为基础，任何无解或多解的题目都是不合格的。

有一天hrdv碰到了一道号称是世界上最难的数独的题目，作为一名合格的程序员，哪能随随便便向困难低头，于是他决定编个程序来解决它。。

输入

第一行有一个数n（0< n <100），表示有n组测试数据，每组测试数据是由一个9*9的九宫格构成，0表示对应的格子为空

输出

输出一个9*9的九宫格，为这个数独的答案

样例输入

1
0 0 5 3 0 0 0 0 0
8 0 0 0 0 0 0 2 0
0 7 0 0 1 0 5 0 0
4 0 0 0 0 5 3 0 0
0 1 0 0 7 0 0 0 6
0 0 3 2 0 0 0 8 0
0 6 0 5 0 0 0 0 9
0 0 4 0 0 0 0 3 0
0 0 0 0 0 9 7 0 0

样例输出

1 4 5 3 2 7 6 9 8
8 3 9 6 5 4 1 2 7
6 7 2 9 1 8 5 4 3
4 9 6 1 8 5 3 7 2
2 1 8 4 7 3 9 5 6
7 5 3 2 9 6 4 8 1
3 6 7 5 4 2 8 1 9
9 8 4 7 6 1 2 3 5
5 2 1 8 3 9 7 6 4 

　　DFS题，记着预处理。

#include <stdio.h>
#include <string.h>
int arr[10][10], count;
bool visrow[10][10], viscol[10][10], vissqu[10][10], ok;
struct Node{
  int x, y;
} tar[82];

int getsqunum(int i, int j){
  return i / 3 * 3 + j / 3;
}

void print(){
  for(int i = 0; i < 9; ++i)
    for(int j = 0; j < 9; ++j)
      if(j != 8) printf("%d ", arr[i][j]);
      else printf("%d\n", arr[i][j]);
}

bool check(int i, int j, int k){
  int t = getsqunum(i, j);
  if(visrow[i][k] || viscol[j][k] || vissqu[t][k]) return 0;
  return 1;
}

void DFS(int k){
  if(k == count){ ok = 1; print(); return; }
  int t = getsqunum(tar[k].x, tar[k].y);
  for(int i = 1; i < 10; ++i){
    if(check(tar[k].x, tar[k].y, i)){
      visrow[tar[k].x][i] = 1;
      viscol[tar[k].y][i] = 1;
      vissqu[t][i] = 1;
      arr[tar[k].x][tar[k].y] = i;

      DFS(k + 1);
      if(ok) return;

      arr[tar[k].x][tar[k].y] = 0;
      visrow[tar[k].x][i] = 0;
      viscol[tar[k].y][i] = 0;
      vissqu[t][i] = 0;
    }
  }
}

int main(){
  int n; scanf("%d", &n);
  while(n--){
    memset(viscol, 0, sizeof(viscol));
    memset(visrow, 0, sizeof(visrow));
    memset(vissqu, 0, sizeof(vissqu));
    count = 0;
    for(int i = 0; i < 9; ++i)
      for(int j = 0; j < 9; ++j){
        scanf("%d", &arr[i][j]);
        if(arr[i][j]){
          visrow[i][arr[i][j]] = 1;
          viscol[j][arr[i][j]] = 1;
          vissqu[getsqunum(i, j)][arr[i][j]] = 1;
        }else{
          tar[count].x = i;
          tar[count++].y = j;
        }
      }
    ok = 0;
    DFS(0);
  }
  return 0;
}