洛谷 P1074 靶形数独 [搜索，水题]

展开
题目描述
小城和小华都是热爱数学的好学生，最近，他们不约而同地迷上了数独游戏，好胜的他们想用数独来一比高低。但普通的数独对他们来说都过于简单了，于是他们向 Z 博士请教，Z 博士拿出了他最近发明的“靶形数独”，作为这两个孩子比试的题目。

靶形数独的方格同普通数独一样，在 99 格宽×99 格高的大九宫格中有9 9 个 33 格宽×33 格高的小九宫格（用粗黑色线隔开的）。在这个大九宫格中，有一些数字是已知的，根据这些数字，利用逻辑推理，在其他的空格上填入 11 到 9 9的数字。每个数字在每个小九宫格内不能重复出现，每个数字在每行、每列也不能重复出现。但靶形数独有一点和普通数独不同，即每一个方格都有一个分值，而且如同一个靶子一样，离中心越近则分值越高。（如图）

上图具体的分值分布是：最里面一格（黄色区域）为 1010 分，黄色区域外面的一圈（红色区域）每个格子为 9 9分，再外面一圈（蓝色区域）每个格子为 88 分，蓝色区域外面一圈（棕色区域）每个格子为 7 7分，最外面一圈（白色区域）每个格子为 6 6分，如上图所示。比赛的要求是：每个人必须完成一个给定的数独（每个给定数独可能有不同的填法），而且要争取更高的总分数。而这个总分数即每个方格上的分值和完成这个数独时填在相应格上的数字的乘积的总和

总分数即每个方格上的分值和完成这个数独时填在相应格上的数字的乘积的总和。如图，在以下的这个已经填完数字的靶形数独游戏中，总分数为 2829。游戏规定，将以总分数的高低决出胜负。

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

输入格式
一共 99 行。每行 9 9个整数（每个数都在 0-90−9 的范围内），表示一个尚未填满的数独方格，未填的空格用“00”表示。每两个数字之间用一个空格隔开。

输出格式
输出共 11 行。输出可以得到的靶形数独的最高分数。如果这个数独无解，则输出整数-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
输出
2829
输入
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
输出
2852

算是八皇后的变种，每次算出来一个ans时要给各个点乘一个权值，比较有意思的是要先给地图的每行缺少的个数排个序，先填空少的，要不然会T；
然后就是写起来比较麻烦。

#include<bits/stdc++.h>
using namespace std;
const int N=10;
int a[N][N],ans[N][N],vis[3][N][N],b[82],maxn,flag;
struct Row{
    int h,zero_cnt;
}row[N];
//排序规则:按照行的0的个数从小到大
int cmp(Row row1,Row row2){
    return row1.zero_cnt<row2.zero_cnt;
}
//获取x，y在哪一个小九宫格子中
int getGrid(int x,int y){
    if(x>=1&&x<=3){
        if(y>=1&&y<=3)  return 1;
        else if(y>=4&&y<=6) return 2;
        else    return 3;
    }
    if(x>=4&&x<=6){
        if(y>=1&&y<=3)  return 4;
        else if(y>=4&&y<=6) return 5;
        else    return 6;
    }
    if(x>=7&&x<=9){
        if(y>=1&&y<=3)  return 7;
        else if(y>=4&&y<=6) return 8;
        else    return 9;
    }
}
//获取x，y位置对应的分数
int getScore(int x,int y){//逐层判断
    if(x==1||y==1||x==9||y==9)   return 6;
    else if(x==2||y==2||x==8||y==8)     return 7;
    else if(x==3||y==3||x==7||y==7)   return 8;
    else if(x==4||y==4||x==6||y==6)   return 9;
    else    return 10;
}
//计算当前组合按照规则计算出来的结果
int cal(){
    int sum=0;
    for(int i=1;i<=9;i++)
        for(int j=1;j<=9;j++)
            sum+=ans[i][j]*getScore(i,j);
    return sum;
}
void dfs(int xh){        //xh:b数组的序号
    if(xh==82){
        flag=1;
        maxn=max(maxn,cal());
        return;
    }
    int x=b[xh]/9+1;    //现在这个序号对应的是第x行，第y列
    int y=b[xh]%9;
    if(y==0)
        x=b[xh]/9,y=9;
    if(!a[x][y]){       //如果当前这个格子是0，才需要填
        for(int j=1;j<=9;j++){  //判断1-9能不能用
            int g=getGrid(x,y);
            if(!vis[0][x][j]&&!vis[1][y][j]&&!vis[2][g][j]){
                ans[x][y]=j;
                vis[0][x][j]=1,vis[1][y][j]=1,vis[2][g][j]=1;
                dfs(xh+1);
                vis[0][x][j]=0,vis[1][y][j]=0,vis[2][g][j]=0;
            }
        }
    }
    else
        dfs(xh+1);
}
void init(){
    for(int i=1;i<=9;i++){
        int cnt=0;                  //cnt:记录该行0的个数
        for(int j=1;j<=9;j++){
            cin>>a[i][j];
            if(a[i][j]==0)
                cnt++;
            else{
                int v=a[i][j];
                int g=getGrid(i,j);
                ans[i][j]=v;       //填答案数组
                vis[0][i][v]=1,vis[1][j][v]=1,vis[2][g][v]=1;//将行列方格的vis标记掉
            }
        }
        row[i].h=i,row[i].zero_cnt=cnt;   //记录每一行有几个0
    }
    sort(row+1,row+1+9,cmp);              //优先安排0少的行
    int num=0;
    for(int i=1;i<=9;i++){
        for(int j=1;j<=9;j++){
            int x=row[i].h,y=j;
            num++;
            b[num]=(x-1)*9+y;           //存储待搜索格子的优先顺序
        }                               //存储一个交换行序以后的搜索顺序的数组
    }
}
int main(){
    init();
    dfs(1);
    if(flag)
        cout<<maxn<<endl;
    else
        cout<<-1<<endl;
    return 0;
}