POJ 2676（数独求解器 搜索）

题目链接poj2676

题目大意

给你一个9*9的数独让你找出一种可行解

分析

题目不难，有点类似于八皇后的做法，放上一个数后，用x、y、z数组标记上

• x[i][k]表示第i列是否能放k(0表示能)
• y[i][k]表示第i行是否能放k
• z[i][k]表示第i个3*3的格子是否能放k 格子标号为0~8

之前一直习惯下标从1开始好理解，但做了这道题之后发现从0开始也是有好处的，从一行的最后一个格子跳到下一行的第一个格子时用取模来做很方便。
总的来说这题的代码写得还算满意

代码

#include<cstdio>
#include<iostream>
#include<cmath>
#include<cstring>
#include<cstdlib>
#include<queue>
#include<stack>
using namespace std;
int T;
int cell[10][10];
int x[10][10];//x[i][k]表示第i列是否能放k
int y[10][10];//y[i][k]表示第i行是否能放k
int z[10][10];//z[i][k]表示第i个3*3的格子是否能放k 格子标号为0~8
int flag;
void print()
{
      for(int i=0;i<9;i++)
      {
            for(int j=0;j<9;j++)
            {
                  cout<<cell[i][j];
            }
            cout<<endl;
      }
}
void Update(int a,int b,int m)//在a,b位置放了x后更新
{
        x[b][m]++;
        y[a][m]++;
        z[b/3+3*(a/3)][m]++;
}
void re_Update(int a,int b,int m)
{
        x[b][m]--;
        y[a][m]--;
        z[b/3+3*(a/3)][m]--;
}
void Init()
{
     memset(x,0,sizeof(x));
     memset(y,0,sizeof(y));
     memset(z,0,sizeof(z));
     flag=0;
     for(int i=0;i<9;i++)
     {
           for(int j=0;j<9;j++)
           {
                 if(cell[i][j]!=0)
                 {
                       Update(i,j,cell[i][j]);
                 }
           }
     }
}
bool Is_out(int a,int b)
{
    if(a>8 || b>8)return 1;
    else return 0;
}
int next_a(int a,int b)//获取a行b列的下一格的行数
{
    if(b+1>8)return a+1;
    else return a;
}
int next_b(int a,int b)
{
      if(b+1>8)return 0;
      else return b+1;
}
bool Is_can(int a,int b,int m)//判断a行b列是否能放x
{
      if(x[b][m]==0 && y[a][m]==0 &&  z[b/3+3*(a/3)][m]==0 )return 1;
      else return 0;
}
void dfs(int a,int b)
{
     if(Is_out(a,b))
     {
           print();
           flag=1;
           return ;
     }
     if(cell[a][b]!=0)
     {
         dfs(next_a(a,b),next_b(a,b));
         if(flag==1)return ;
     }
     else
     {
           for(int i=1;i<=9;i++)
           {
                 if(Is_can(a,b,i))
                 {
                       cell[a][b]=i;
                       Update(a,b,i);
                       dfs(next_a(a,b),next_b(a,b));
                       cell[a][b]=0;
                       re_Update(a,b,i);
                       if(flag==1)return ;
                 }
           }
     }
}
int main()
{
    scanf("%d",&T);
    while(T--)
    {
          char c;
          for(int i=0;i<9;i++)
                for(int j=0;j<9;j++)
                {
                    scanf("%c",&c);
                    while((int)c<'0' || (int )c>'9')scanf("%c",&c);
                    cell[i][j]=(int)c-(int)'0';
                }
          //print();
          Init();
          dfs(0,0);
    }
    return 0;
}