金字塔数独

问题描述

蒜头君天资聪颖，酷爱数学，尤其擅长做数独游戏。不过普通的数独游戏已经满足不了蒜头君了，于是他发明了一种“金字塔数独”：
下图即为金字塔数独。和普通数独一样，在 9 × 9 的大九宫格中有 9 个 3 ×3 的小九宫格（用粗黑色线隔开的）。要求每个格子上都有一个 1 到 9 的数字，每个数字在每个小九宫格内不能重复出现，每个数字在每行、每列也不能重复出现。
但金字塔数独的每一个格子都有一个分值，类似金字塔的俯视图。如图所示，白色格子为 6 分，蓝色格子为 7 分，绿色格子为 8 分，紫色格子为 9 分，红色格子为 10 分。颜色相同的格子分值一样，离中心越近则分值越高。

金字塔数独的总分等于每个格子上的数字和对应的分值乘积之和。现在蒜头君给定金字塔数独的若干数字，请问如何填写，可以使得金字塔数独的总分最高。
输入格式
输入一共 9 行。每行输入 9 个整数（每个数都在 0—9 的范围内），每两个整数之间用一个空格隔开，“0”表示该格子为空。
输出格式
输出为一行，输出一个整数，代表金字塔数独的最高总分。如果数独无解，则输出 −1。
样例输入
0 0 0 0 0 0 0 0 0
0 0 3 0 0 0 9 0 0
7 9 0 0 0 2 0 1 3
0 0 9 1 5 0 0 0 0
0 7 4 0 2 6 1 3 9
0 0 6 0 0 0 0 0 0
6 0 0 0 0 7 0 0 0
3 1 0 4 0 5 7 9 6
0 0 7 0 0 1 0 4 0
样例输出

2864

代码:

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
int a[10][10],hang[10][10],lie[10][10],room[10][10];
int mx=0;
int goal=0;int kong=0;
struct hnum{
	int h,num;
	bool operator <(const hnum &A) const{
		return num<A.num;
	}
}h0[10];
int wh_room(int i,int j){
	if(i<=3){
		if(j<=3)return 1;
		else if(j>3&&j<=6)return 2;
		else return 3;
	}
	else if(i>3&&i<=6){
		if(j<=3)return 4;
		else if(j>3&&j<=6)return 5;
		else return 6;
	}
	else {
		if(j<=3)return 7;
		else if(j>3&&j<=6)return 8;
		else return 9;
	}
}
int value(int i,int j){
	if(i==1||j==1||i==9||j==9)   return 6;
    if(i==2||j==2||i==8||j==8)     return 7;
    if(i==3||j==3||i==7||j==7)   return 8;
    if(i==4||j==4||i==6||j==6)   return 9;
    return 10;
}
void dfs(int x,int zero,int much,int jj,int k){
	if(much==kong){
		if(mx<goal)mx=goal;
		return;
	}
	if(zero==0){
		k++;
		x=h0[k].h;zero=h0[k].num;
		jj=0; 
	}
	for(int j=jj+1;j<=9;j++){
		if(!a[x][j]){
			for(int i=1;i<=9;i++){
				if(!hang[x][i]&&!lie[j][i]&&!room[wh_room(x,j)][i]){
					hang[x][i]=1;lie[j][i]=1;room[wh_room(x,j)][i]=1;
					goal+=i*value(x,j);
					dfs(x,zero-1,much+1,j,k);
					goal-=i*value(x,j);
					hang[x][i]=0;lie[j][i]=0;room[wh_room(x,j)][i]=0;
				}
			}
		return;
	    }
    }
}
int main(){
	int ip1;
	for(int i=1;i<=9;i++)
	 for(int j=1;j<=9;j++){
	 	 scanf("%d",&ip1);
	 	 if(ip1==0){
	 	 	h0[i].h=i;
	 	 	h0[i].num++;
	 	 	kong++;
		  }
	 	 else {
	 	 	a[i][j]=ip1;
	 	 	goal+=ip1*value(i,j);
	 	 	hang[i][ip1]=1;lie[j][ip1]=1;
	 	 	room[wh_room(i,j)][ip1]=1;
		  }
	 }
	 sort(h0+1,h0+10);
	 dfs(h0[1].h,h0[1].num,0,0,1);
	 printf("%d",mx);
	 return 0;
}