洛谷 P1074[NOIP2009 提高组] 靶形数独题解（搜索+剪枝优化）

最新推荐文章于 2024-07-25 11:38:39 发布

EllinY(垂耳兔)

最新推荐文章于 2024-07-25 11:38:39 发布

阅读量1.1k

点赞数 17

文章标签：剪枝算法深度优先 c++ 笔记

本文链接：https://blog.csdn.net/2401_84512298/article/details/140211941

版权

题意

没玩过数独的先看一下规则 ~~（应该没人没玩过吧）~~

数独盘面是九个宫（以粗实线划分），每一宫又分为九个小格。在空格上填入1-9的数字，使1-9每个数字在每一行、每一列和每一宫中都只出现一次。

你需要填一个数独，但有一点和普通的不一样，你要使填好后数独的分数最高。每个格子获得的分数是这个格子本身的分值×格子上填的数，整个数独的分数是每个格子的分数之和。每个格子本身的分值如下图，用颜色区分不同区域：

如图，在以下的这个已经填完数字的靶形数独游戏中，总分数为 $2829$ 。（下图也是样例的最优解）

样例输入

7 0 0 9 0 0 0 0 1 
1 0 0 0 0 5 9 0 0 
0 0 0 2 0 0 0 8 0 
0 0 5 0 2 0 0 0 3 
0 0 0 0 0 0 6 4 8 
4 1 3 0 0 0 0 0 0 
0 0 7 0 0 2 0 9 0 
2 0 1 0 6 0 8 0 4 
0 8 0 5 0 4 0 1 2

样例输出

数据规模

对于 $40\%$ 的数据，数独中非 $0$ 数的个数不少于 $30$ ；
对于 $80\%$ 的数据，数独中非 $0$ 数的个数不少于 $26$ ；
对于 $100\%$ 的数据，数独中非 $0$ 数的个数不少于 $24$ 。

思路分析

阶段1

可以想到一个较为简单的做法，就是用dfs枚举每个空格填什么，全部填完之后再判断每行、每列、每个宫格是否有重复的，如果没有，就算出分数计入答案。
时间复杂度：最多有 $57$ 个空格，每个空格有9种填法，也就是9^57，实在是太恐怖了，计算器都不给我一个确切的答案。

阶段2

要避免一些肯定不对的情况，就要一边填，一边判断当前这一行，这一列，这一宫格有没有出现重复的数字，可以建立 $b oo l$ 数组 $f r [i] [j]$ ， $f c [i] [j]$ ， $f [i] [j]$ ，分别表示第 $i$ 行，第 $i$ 列，第 $i$ 个宫格是否已经有数字 $j$ 了
代码如下：（ $g e t i d (x, y)$ 是用来获取第 $x$ 行第 $y$ 列的格子是属于哪个宫格的）

for(int i=1;i<=9;i++)
		if(!fr[x][i]&&!fc[y][i]&&!f[getid(x,y)][i]){
			fr[x][i]=1,fc[y][i]=1,f[getid(x,y)][i]=1;
			rec[x][y]=i;
			if(y==9) dfs(x+1,1);
			else dfs(x,y+1);
			fr[x][i]=0,fc[y][i]=0,f[getid(x,y)][i]=0;
		}

阶段3

然而上面的代码只有80分，于是考虑用一些 ~~玄学卡常~~ 优化。我们统计出每一行有多少个 $0$ ，并将这些行按照 $0$ 的个数从小到大排序，先填 $0$ 个数少的行（ $df s$ 本来就是一行行填入的）。这样做能使枚举的情况数量变少，到后面有很多未知的行的时候，已经填好了很多行，可以剪掉很多情况。
由于打乱原本的数组会导致所属的宫格无法判断，所以我用一个 $s t r u c t$ 存了每行的编号 $i d$ 和 $0$ 的数量 $c n t$ ，单独排好序，用 $h e a d$ 存储 $0$ 最少的是第几行， $n e x [i]$ 记录按照 $0$ 的数量排序后第 $i$ 行的下面一行是第几行，形成一个记录顺序的链表。
完整的代码奉上：（只要用 $n e x [x]$ 替换之前的 $x + 1$ 即可）

#include<bits/stdc++.h>
using namespace std;
int a[10][10],rec[10][10],ans,head,nex[10];
bool fr[10][10],fc[10][10],f[10][10];
int score[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}};
 //每个格子的分值我采用了打表，可能有更好的方法
struct node{
	int id,cnt;
} n[10];
bool cmp(node x,node y){
	return x.cnt<y.cnt;
}
int getid(int i,int j){
	if(i<=3){
		if(j<=3) return 1;
		else if(j>=7) return 3;
		else return 2;
	}
	else if(i>=7){
		if(j<=3) return 7;
		else if(j>=7) return 9;
		else return 8;
	}
	else{
		if(j<=3) return 4;
		else if(j>=7) return 6;
		else return 5;
	}
}
void dfs(int x,int y){
	if(x==0){
		int res=0;
		for(int i=1;i<=9;i++)
			for(int j=1;j<=9;j++)
				res+=rec[i][j]*score[i][j];
		ans=max(ans,res);
		return;
	}
	if(a[x][y]){
		rec[x][y]=a[x][y];
		if(y==9) dfs(nex[x],1);
		else dfs(x,y+1);
	}
	else{
		for(int i=1;i<=9;i++)
			if(!fr[x][i]&&!fc[y][i]&&!f[getid(x,y)][i]){
				fr[x][i]=1,fc[y][i]=1,f[getid(x,y)][i]=1;
				rec[x][y]=i;
				if(y==9) dfs(nex[x],1);
				else dfs(x,y+1);
				fr[x][i]=0,fc[y][i]=0,f[getid(x,y)][i]=0;
			}
	}
}
int main(){
	for(int i=1;i<=9;i++) n[i].id=i;
	for(int i=1;i<=9;i++)
		for(int j=1;j<=9;j++){
			scanf("%d",&a[i][j]);
			n[i].cnt+=(a[i][j]==0);
			fr[i][a[i][j]]=1;
			fc[j][a[i][j]]=1;
			f[getid(i,j)][a[i][j]]=1;
		}
	sort(n+1,n+10,cmp);
	head=n[1].id;
	for(int i=1;i<9;i++) nex[n[i].id]=n[i+1].id;
	dfs(head,1);
	if(ans==0) printf("-1\n");
	else printf("%d\n",ans);
	return 0;
}