POJ 2676 挺复杂的一道深搜

 1 #include<iostream>
 2 #include<cstring>
 3 #include<vector>
 4 #include<algorithm>
 5 using namespace std;
 6 short row[9][10],col[9][10],block[9][10];//rowFlag[i][num]=1表示在第i行已经放了数字num，后2个分别表示列、块
 7 int board[9][10];
 8 struct pos
 9 {
10     int x,y;
11 }blank[81];
12 int Getnum(int x,int y)//状态压缩
13 {
14     int xx=x/3;
15     int yy=y/3;
16     return xx*3+yy;
17 }
18 void setallflag(int x,int y,int num,int f)
19 {
20     row[x][num]=f;
21     col[y][num]=f;
22     block[Getnum(x,y)][num]=f;
23 }
24 bool isok(int x,int y,int num)
25 {
26     return !row[x][num]&&!col[y][num]&&!block[Getnum(x,y)][num];
27 }
28 bool dfs(int n)
29 {
30     if(n<0) return true;
31     int x=blank[n].x;
32     int y=blank[n].y;
33     for(int num=1;num<=9;num++)
34     {
35         if(isok(x,y,num))
36         {
37             board[x][y]=num;
38             setallflag(x,y,num,1);
39             if(dfs(n-1))
40                 return true;
41             setallflag(x,y,num,0);
42         }
43     }
44     return false;
45 }
46 int main()
47 {
48     int t;
49     cin>>t;
50     while(t--)
51     {
52         memset(row,0,sizeof(row));
53         memset(col,0,sizeof(col));
54         memset(block,0,sizeof(block));
55         memset(blank,0,sizeof(blank));
56         int flag=0;
57         for(int i=0;i<9;i++)
58             for(int j=0;j<9;j++)
59         {
60             char c;
61             cin>>c;
62             board[i][j]=c-‘0‘;
63             if(board[i][j])
64                 setallflag(i,j,board[i][j],1);
65             else
66             {
67                 blank[flag].x=i;
68                 blank[flag].y=j;
69                 flag++;
70             }
71         }
72         if(dfs(flag-1))
73         {
74             for(int i=0;i<9;i++)
75             {
76                 for(int j=0;j<9;j++)
77                     cout<<board[i][j];
78                 cout<<endl;
79             }
80         }
81     }
82     return 0;
83 }