HDU5547-Sodoku 数独 DFS

点击打开链接

题意：输入一个由1,2,3,4*组成的4*4的板块，板块再分成2*2个宫，每个宫由2*2个格子组成，要求同一行，同一列，同一宫中不能出现相同的数字。

思路：数据不大，直接DFS

#include<cstdio>
using namespace std;
char a[8][8];
struct node
{
    int x;
    int y;
};
node q[20];
int num;
bool yes(int x,int y)
{
    int sum;
    if(x<=2&&y<=2)
    {
        sum=0;
        if(a[1][1]==a[x][y]) sum++;
        if(a[1][2]==a[x][y]) sum++;
        if(a[2][1]==a[x][y]) sum++;
        if(a[2][2]==a[x][y]) sum++;
        if(sum>=2) return 0;
    }
    else if(x<=2&&y>2)
    {
        sum=0;
        if(a[1][3]==a[x][y]) sum++;
        if(a[1][4]==a[x][y]) sum++;
        if(a[2][3]==a[x][y]) sum++;
        if(a[2][4]==a[x][y]) sum++;
        if(sum>=2) return 0;
    }
    else if(x>2&&y<=2)
    {
        sum=0;
        if(a[3][1]==a[x][y]) sum++;
        if(a[3][2]==a[x][y]) sum++;
        if(a[4][1]==a[x][y]) sum++;
        if(a[4][2]==a[x][y]) sum++;
        if(sum>=2) return 0;
    }
    else
    {
        sum=0;
        if(a[3][3]==a[x][y]) sum++;
        if(a[3][4]==a[x][y]) sum++;
        if(a[4][3]==a[x][y]) sum++;
        if(a[4][4]==a[x][y]) sum++;
        if(sum>=2) return 0;
    }

    sum=0;
    if(a[x][1]==a[x][y]) sum++;
    if(a[x][2]==a[x][y]) sum++;
    if(a[x][3]==a[x][y]) sum++;
    if(a[x][4]==a[x][y]) sum++;
    if(sum>=2) return 0;

    sum=0;
    if(a[1][y]==a[x][y]) sum++;
    if(a[2][y]==a[x][y]) sum++;
    if(a[3][y]==a[x][y]) sum++;
    if(a[4][y]==a[x][y]) sum++;
    if(sum>=2) return 0;

    return 1;
}
void dfs(int i)
{
    if(i>=num)
    {
        for(int k=1;k<=4;k++)
            printf("%s\n",a[k]+1);
        return;
    }
    for(int j=1;j<=4;j++)
    {
        a[q[i].x][q[i].y]=(char)(j+'0');
        if(yes(q[i].x,q[i].y))
            dfs(i+1);
        a[q[i].x][q[i].y]='*';
    }
}
int main()
{
    int T,cas=1;
    int i,j;
    scanf("%d",&T);
    while(T--)
    {
        for(i=1;i<=4;i++)
        {
            scanf("%s",a[i]+1);
        }
        num=0;
        for(i=1;i<=4;i++)
        {
            for(j=1;j<=4;j++)
            {
                if(a[i][j]=='*')
                {
                    q[num].x=i;
                    q[num].y=j;
                    num++;
                }
            }
        }
        printf("Case #%d:\n",cas++);
        dfs(0);
    }
    return 0;
}