HJ44 Sudoku

真开心，之前写过华为的数独题，直接提交就OK了=v=

#include<bits/stdc++.h>
using namespace std;
int a[10][10],mp[10][10][10],cnt_mp[10][10],num_sta=0,ans[10][10];
bool b[10],fg[10],fd=0;
struct Sta{
    int x,y;
}sta[100];
struct ID{
    int x,y;
}id[10][10];
void Dfs(int k){
    if(k>num_sta){
        fd=1;
        for(int i=1;i<=9;i++)
            for(int j=1;j<=9;j++)
                ans[i][j]=a[i][j];
        return;
    }
    if(fd)return;
    int x=sta[k].x,y=sta[k].y;
    for(int i=1;i<=cnt_mp[x][y];i++){
        if(fd)return;
        int wl=mp[x][y][i];
        for(int j=1;j<=9;j++)fg[j]=1;
        for(int j=1;j<=9;j++)if(j!=y)fg[a[x][j]]=0;
        if(fg[wl]==0) continue;
        for(int j=1;j<=9;j++)fg[j]=1;
        for(int j=1;j<=9;j++)if(j!=x)fg[a[j][y]]=0;
        if(fg[wl]==0) continue;
        for(int j=1;j<=9;j++)fg[j]=1;
        for(int dx=-1;dx<=1;dx++){
            for(int dy=-1;dy<=1;dy++){
                fg[a[id[x][y].x+dx][id[x][y].y+dy]]=0;
            }
        }
        if(fg[wl]==0) continue;
        a[x][y]=wl;
        Dfs(k+1);
        a[x][y]=0;
    }
    return;
}
int main(){
    for(int i=1;i<=9;i++){
        for(int j=1;j<=9;j++){
            scanf("%d",&a[i][j]);
            id[i][j].x=(((i+2)/3)-1)*3+2;
            id[i][j].y=(((j+2)/3)-1)*3+2;
        }
    }
    for(int i=1;i<=9;i++){
        for(int j=1;j<=9;j++)
            if(a[i][j]==0){
                sta[++num_sta].x=i;
                sta[num_sta].y=j;
                for(int k=1;k<=9;k++)b[k]=1;
                for(int k=1;k<=9;k++){
                    if(a[i][k]) b[a[i][k]]=0;
                    if(a[k][j]) b[a[k][j]]=0;
                }
                for(int dx=-1;dx<=1;dx++){
                    for(int dy=-1;dy<=1;dy++){
                        b[a[id[i][j].x+dx][id[i][j].y+dy]]=0;
                    }
                }
                for(int k=1;k<=9;k++)
                    if(b[k]){
                        cnt_mp[i][j]++;
                        mp[i][j][cnt_mp[i][j]]=k;
                    }
            }
    }
    Dfs(1);
    for(int i=1;i<=9;i++){
        for(int j=1;j<=9;j++)printf("%d ",ans[i][j]);
        puts("");
    }
    return 0;
}