洛谷OJ 1074 靶型sudoku dfs(搜索顺序优化)

#include <bits/stdc++.h> 
using namespace std;
const int N=2e2+20;
int row[N][N],col[N][N],vis[N][N];//row[i][1~9] 第i列是否能选j 
int a[N][N]; 
int ans,n=9,cnt;
int dist(int x,int y)
{
    return (x/3)*3+(y/3);
}
bool check(int x,int y,int k)
{
    if(row[x][k]||col[y][k])
        return false;
    if(vis[dist(x,y)][k])
        return false;
    return true;
} 
int b[][9]={
{6,6,6,6,6,6,6,6,6},
{6,7,7,7,7,7,7,7,6},
{6,7,8,8,8,8,8,7,6},
{6,7,8,9,9,9,8,7,6},
{6,7,8,9,10,9,8,7,6},
{6,7,8,9,9,9,8,7,6},
{6,7,8,8,8,8,8,7,6},
{6,7,7,7,7,7,7,7,6},
{6,6,6,6,6,6,6,6,6},
};
struct node{
    int x,y;
};
node getnext()  //找到可能性小的 
{  
    node tmp;  
    int min,t;  
    tmp.x=n;  
    tmp.y=n;  
    min=n;  
    for (int i=0;i<9;i++)  
        for (int j=0;j<9;j++)   
            if(!a[i][j])  
            {  
                t=0;  
                for (int k=1;k<=9;k++)  
                    if (!row[i][k])  
                        if (!col[j][k])  
                            if (!vis[dist(i,j)][k]) 
                                t++;  
                if (t<min)  
                {  
                    min=t;  
                    tmp.x=i;  
                    tmp.y=j;  
                }  
            }  
    return tmp;  
}  
void dfs(int p)
{    
    if(p>=cnt)//需要填cnt个 
    {
        int res=0;
        for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
                res+=b[i][j]*a[i][j];
        ans=max(ans,res);
        return;
    }
    node tmp=getnext();
    if(tmp.x==n&&tmp.y==n)
        return;
    int x=tmp.x,y=tmp.y;
    for(int k=1;k<=9;k++)
    {
        if(check(x,y,k))
        {
            row[x][k]=col[y][k]=vis[dist(x,y)][k]=1;
            a[x][y]=k;
            dfs(p+1);
            row[x][k]=col[y][k]=vis[dist(x,y)][k]=0;
            a[x][y]=0;
        }
    }
}
int main()
{
    int n=9;
    ans=-1;
    cnt=n*n;
    memset(vis,0,sizeof(vis));
    memset(col,0,sizeof(col));
    memset(row,0,sizeof(row));
    for(int i=0;i<9;i++)
    {
        for(int j=0;j<9;j++)
        {
            scanf("%d",&a[i][j]);
            if(a[i][j])
            {
                row[i][a[i][j]]=1;
                col[j][a[i][j]]=1;
                vis[dist(i,j)][a[i][j]]=1;
                cnt--;//
            }
        }
    }
    dfs(0);
    cout<<ans<<endl;
    return 0;    
}