问题描述
蒜头君天资聪颖,酷爱数学,尤其擅长做数独游戏。不过普通的数独游戏已经满足不了蒜头君了,于是他发明了一种“金字塔数独”:
下图即为金字塔数独。和普通数独一样,在 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;
}