题目来源:https://www.luogu.org/problem/show?pid=1074#sub
相当繁琐的dfs,TLE了许多次,终于AC了。
状态压缩来记录每个格子所在的行、列、区域内1~9的数字是否填过。(也可采用类似八皇后问题的处理方法)通过位运算操作。
分别用h、l、b记录每一行、每一列、每一区域当前的状态,二进制下往左数第x位上是1表示数x再该行(列、区域)内已填过。
重要的剪枝:
先填可以填的数字个数少的格子,再填可填数字个数多的,这样可减少dfs的宽度。
代码:
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
int g[10][10]={{0,0,0,0,0,0,0,0,0,0},{0,6,6,6,6,6,6,6,6,6},{0,6,7,7,7,7,7,7,7,6},{0,6,7,8,8,8,8,8,7,6},{0,6,7,8,9,9,9,8,7,6},{0,6,7,8,9,10,9,8,7,6},{0,6,7,8,9,9,9,8,7,6},{0,6,7,8,8,8,8,8,7,6},{0,6,7,7,7,7,7,7,7,6},{0,6,6,6,6,6,6,6,6,6}};//该所在格子的分数
int a[10][10];
int in[10][10]={{0,0,0,0,0,0,0,0,0,0},{0,1,1,1,2,2,2,3,3,3},{0,1,1,1,2,2,2,3,3,3},{0,1,1,1,2,2,2,3,3,3},{0,4,4,4,5,5,5,6,6,6},{0,4,4,4,5,5,5,6,6,6},{0,4,4,4,5,5,5,6,6,6},{0,7,7,7,8,8,8,9,9,9},{0,7,7,7,8,8,8,9,9,9},{0,7,7,7,8,8,8,9,9,9}};//该所在格子所属的区域
int l[10],h[10],b[10];
int cnt=0,tot=-1;
int sx[100],sy[100];
void dfs(int v)
{
int x=0,y=0,num=0;
for(int i=1;i<=cnt;i++)
{
if(a[sx[i]][sy[i]]==0)
{
int t=0;
int c=(h[sx[i]]|l[sy[i]]|b[in[sx[i]][sy[i]]]);
for(int k=1;k<=9;k++)if(((1<<k)&c)!=0)t++;
if(t>num)
{
num=t;
x=sx[i];
y=sy[i];
}
}
}
if(x==0)
{
tot=max(tot,v);
return;
}
int s=(h[x]|l[y]|b[in[x][y]]);
if(s==1022)return;
for(int k=1;k<=9;k++)
{
if((s&(1<<k))==0)
{
a[x][y]=k;
h[x]=(h[x]|(1<<k));l[y]=(l[y]|(1<<k));
b[in[x][y]]=(b[in[x][y]]|(1<<k));
dfs(v+g[x][y]*a[x][y]);
a[x][y]=0;
h[x]=(h[x]^(1<<k));l[y]=(l[y]^(1<<k));
b[in[x][y]]=(b[in[x][y]]^(1<<k));
}
}
}
int main()
{
ios::sync_with_stdio(false);
int s=0;
for(int i=1;i<=9;i++)
{
for(int j=1;j<=9;j++)
{
cin>>a[i][j];
if(a[i][j]!=0)
{
int c=(1<<a[i][j]);
h[i]=(h[i]|c);
l[j]=(l[j]|c);
b[in[i][j]]=(b[in[i][j]]|c);
s+=g[i][j]*a[i][j];
}
else
{
sx[++cnt]=i;
sy[cnt]=j;
}
}
}
dfs(s);
cout<<tot;
return 0;
}