bzoj 3109: [cqoi2013]新数独【dfs】

按3x3的小块dfs，填数的时候直接满足所有条件即可

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=15;
int a[N][N],ans[N][N];
char b[N][N],c[N][N];
bool v[5][5][N],h[N][N],l[N][N];
char read()
{
    char p=getchar();
    while(p!='^'&&p!='v'&&p!='<'&&p!='>')
        p=getchar();//cerr<<p<<endl;
    return p;
}
bool ok(int x,int y,int i)
{//cerr<<b[x][y-1]<<" "<<x<<" "<<y<<" "<<i<<"   "<<a[x][y-1]<<endl;
    if(h[x][i]||l[y][i])
        return 0;
    if(y%3!=1)
    {
        if(b[x][y-1]=='>'&&a[x][y-1]<i)
            return 0;
        if(b[x][y-1]=='<'&&a[x][y-1]>i)
            return 0;
    }
    if(x%3!=1)
    {
        if(c[x-1][y]=='^'&&a[x-1][y]>i)
            return 0;
        if(c[x-1][y]=='v'&&a[x-1][y]<i)
            return 0;
    }
    return 1;
}
void dfs(int p,int q,int x,int y)
{
    if(ans[1][1])
        return;
    if(p==3)
    {
        for(int i=1;i<=9;i++)
            for(int j=1;j<=9;j++)
                ans[i][j]=a[i][j];
        return;
    }
    for(int i=1;i<=9;i++)
        if(!v[p][q][i]&&ok(p*3+x,q*3+y,i))
        {
            v[p][q][i]=1,h[p*3+x][i]=1,l[q*3+y][i]=1,a[p*3+x][q*3+y]=i;
            if(q==2&&x==3&&y==3)
                dfs(p+1,0,1,1);
            else if(x==3&&y==3)
                dfs(p,q+1,1,1);
            else if(y==3)
                dfs(p,q,x+1,1);
            else
                dfs(p,q,x,y+1);
            v[p][q][i]=0,h[p*3+x][i]=0,l[q*3+y][i]=0,a[p*3+x][q*3+y]=0;
        }
}
int main()
{
    for(int i=0;i<3;i++)
    {
        for(int j=0;j<6;j++)
            b[i*3+1][j/2*3+j%2+1]=read();
        for(int j=1;j<=9;j++)
            c[i*3+1][j]=read();
        for(int j=0;j<6;j++)
            b[i*3+2][j/2*3+j%2+1]=read();
        for(int j=1;j<=9;j++)
            c[i*3+2][j]=read();
        for(int j=0;j<6;j++)
            b[i*3+3][j/2*3+j%2+1]=read();
    }
    // for(int i=1;i<=9;i++)
    // {
        // for(int j=1;j<=9;j++)
            // cerr<<c[i][j];
        // cerr<<endl;
    // }
    dfs(0,0,1,1);
    for(int i=1;i<=9;i++)
    {
        for(int j=1;j<=9;j++)
            printf("%d ",ans[i][j]);
        puts("");
    }
    return 0;
}

转载于:https://www.cnblogs.com/lokiii/p/9737219.html