靶形数独【NOIP2009提高组】

题目描述

　　小城和小华都是热爱数学的好学生，最近，他们不约而同地迷上了数独游戏，好胜的他们想用数独来一比高低。但普通的数独对他们来说都过于简单了，于是他们向Z 博士请教，Z 博士拿出了他最近发明的“靶形数独”，作为这两个孩子比试的题目。靶形数独的方格同普通数独一样，在 9 格宽×9 格高的大九宫格中有9 个3 格宽×3 格高的小九宫格（用粗黑色线隔开的）。在这个大九宫格中，有一些数字是已知的，根据这些数字，利用逻辑推理，在其他的空格上填入1 到9 的数字。每个数字在每个小九宫格内不能重复出现，每个数字在每行、每列也不能重复出现。但靶形数独有一点和普通数独不同，即每一个方格都有一个分值，而且如同一个靶子一样，离中心越近则分值越高。（如图）

　　上图具体的分值分布是：最里面一格（黄色区域）为 10 分，黄色区域外面的一圈（红色区域）每个格子为9 分，再外面一圈（蓝色区域）每个格子为8 分，蓝色区域外面一圈（棕色区域）每个格子为7 分，最外面一圈（白色区域）每个格子为6 分，如上图所示。比赛的要求是：每个人必须完成一个给定的数独（每个给定数独可能有不同的填法），而且要争取更高的总分数。而这个总分数即每个方格上的分值和完成这个数独时填在相应格上的数字的乘积的总和。如图，在以下的这个已经填完数字的靶形数独游戏中，总分数为2829。游戏规定，将以总分数的高低决出胜负。

　　由于求胜心切，小城找到了善于编程的你，让你帮他求出，对于给定的靶形数独，能够得到的最高分数。

输入输出格式

输入格式：

　　输入文件名为 sudoku.in。
　　一共 9 行。每行9 个整数（每个数都在0—9 的范围内），表示一个尚未填满的数独方格，未填的空格用“0”表示。每两个数字之间用一个空格隔开。

输出格式：

　　输出文件 sudoku.out 共1 行。
　　输出可以得到的靶形数独的最高分数。如果这个数独无解，则输出整数-1。

输入输出样例

输入样例#1：

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

输出样例#1：

2829

提示信息

【输入输出样例 2】
sudoku.in
0 0 0 7 0 2 4 5 3
9 0 0 0 0 8 0 0 0
7 4 0 0 0 5 0 1 0
1 9 5 0 8 0 0 0 0
0 7 0 0 0 0 0 2 5
0 3 0 5 7 9 1 0 8
0 0 0 6 0 1 0 0 0
0 6 0 9 0 0 0 0 1
0 0 0 0 0 0 0 0 6

sudoku.out
2852

【数据范围】

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

算法分析：

对于填写数独这种非常非常经典的问题来说，

有一种非常非常经典的算法：DFS

首先，时间复杂度判断是否可行。

81个格子，9种情况。

最裸最裸的DFS也无非O(9^81)

但稍微接触过一点深搜的人都会用一种很简单的可行性剪枝，即判断该行，该列是否有不能填的数字。若有，则不填：

事先存好每个位置的得分：

int ma2[11][11]={
	{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},
};

然后按深搜的套路模拟：

#include<bits/stdc++.h>
using namespace std;
int a[12][12];//a[i][j]表示第j个数是否出现在第i行
int b[12][12];//b[i][j]表示第j个数是否出现在第i列
int c[4][4][12];//c[i][j][k]表示第k个数是否出现在第i行第j列的3*3方块中
int ma[12][12],maxn=0,change=0,maxn9=0;//ma即数独，maxn即最大得分，maxn9为一个小剪枝
int ma2[11][11]={
	{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},
};
void dfs(int x,int y,int val)//（x,y）为要填数的坐标，val为当前最大分数
{
	if(x==9&&y==1)
	{
		if(val>maxn9) maxn9=val;
		else return;//第9行得分的因数相同，最优性剪枝
	}
	if(ma[x][y]) 
	{
		if(y==9) dfs(x+1,1,val+ma[x][y]*ma2[x-1][y-1]);
		else dfs(x,y+1,val+ma[x][y]*ma2[x-1][y-1]);
		return;
	}//当前位置已经有数字，直接找下一个
	if(x==10)
	{
		maxn=max(maxn,val);
		return;
	}
	for(int i=9;i>=1;i--)
	{
		if(a[x][i]) continue;
		if(b[y][i]) continue;
		if(c[(x-1)/3+1][(y-1)/3+1][i]) continue;//可行性剪枝
		ma[x][y]=i;
		a[x][i]=b[y][i]=c[(x-1)/3+1][(y-1)/3+1][i]=1;//标记
		if(y==9) dfs(x+1,1,val+ma[x][y]*ma2[x-1][y-1]);
		else dfs(x,y+1,val+ma[x][y]*ma2[x-1][y-1]);
		a[x][i]=b[y][i]=c[(x-1)/3+1][(y-1)/3+1][i]=0;//回溯
		ma[x][y]=0;
	}
}
int main()
{
	for(int i=1;i<=9;i++)
	{
		for(int j=1;j<=9;j++)
		{
			int x;
			cin>>x;
			ma[i][j]=x;
			if(x)
			{
				a[i][x]=2;
				b[j][x]=2;
				c[(i-1)/3+1][(j-1)/3+1][x]=2;
			}
		}
	}
	dfs(1,1,0);
	if(maxn)cout<<maxn;
	else cout<<-1;
	return 0; 
}

但......仍旧会超时（15/20）

那么，一种新的剪枝出现了，填数独的时候一般会从已知数少的那行填起。

于是改变填数顺序后，可以完美AC此题。

#include<bits/stdc++.h>
using namespace std;
int a[12][12];
int b[12][12];
int c[4][4][12];
struct N{
	int q;
	int index;
}p[12];
int ma[12][12],maxn=0;
int ma2[11][11]={
	{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},
};
int cmp(N x,N y)
{
	return x.q<y.q;
}
void dfs(int x,int y,int val)
{
	int xr=p[x].index;
	if(ma[xr][y]) 
	{
		if(y==9) dfs(x+1,1,val+ma[xr][y]*ma2[xr-1][y-1]);
		else dfs(x,y+1,val+ma[xr][y]*ma2[xr-1][y-1]);
		return;
	}
	if(x==10)
	{
		maxn=max(maxn,val);
		return;
	}
	for(int i=9;i>=1;i--)
	{
		if(a[xr][i]) continue;
		if(b[y][i]) continue;
		if(c[(xr-1)/3+1][(y-1)/3+1][i]) continue;
		ma[xr][y]=i;
		a[xr][i]=b[y][i]=c[(xr-1)/3+1][(y-1)/3+1][i]=1;
		if(y==9) dfs(x+1,1,val+ma[xr][y]*ma2[xr-1][y-1]);
		else dfs(x,y+1,val+ma[xr][y]*ma2[xr-1][y-1]);
		a[xr][i]=b[y][i]=c[(xr-1)/3+1][(y-1)/3+1][i]=0;
		ma[xr][y]=0;
	}
}
int main()
{
	for(int i=1;i<=9;i++)
	{
		for(int j=1;j<=9;j++)
		{
			int x;
			cin>>x;
			ma[i][j]=x;
			if(!x) p[i].q++,p[i].index=i;
			if(x)
			{
				a[i][x]=2;
				b[j][x]=2;
				c[(i-1)/3+1][(j-1)/3+1][x]=2;
			}
		}
	}
	sort(p+1,p+10,cmp);
	dfs(1,1,0);
	if(maxn)cout<<maxn;
	else cout<<-1;
	return 0; 
}