luogu P1074 靶形数独

P1074 靶形数独

题目描述

小城和小华都是热爱数学的好学生，最近，他们不约而同地迷上了数独游戏，好胜的他们想用数独来一比高低。但普通的数独对他们来说都过于简单了，于是他们向 Z 博士请教，Z 博士拿出了他最近发明的“靶形数独”，作为这两个孩子比试的题目。
靶形数独的方格同普通数独一样，在 9 格宽×9 格高的大九宫格中有 9 个 3 格宽×3 格高的小九宫格（用粗黑色线隔开的）。在这个大九宫格中，有一些数字是已知的，根据这些数字，利用逻辑推理，在其他的空格上填入 11 到 9的数字。每个数字在每个小九宫格内不能重复出现，每个数字在每行、每列也不能重复出现。但靶形数独有一点和普通数独不同，即每一个方格都有一个分值，而且如同一个靶子一样，离中心越近则分值越高。（如图）
上图具体的分值分布是：最里面一格（黄色区域）为 10 分，黄色区域外面的一圈（红色区域）每个格子为 9分，再外面一圈（蓝色区域）每个格子为 8 分，蓝色区域外面一圈（棕色区域）每个格子为 7分，最外面一圈（白色区域）每个格子为 6分，如上图所示。比赛的要求是：每个人必须完成一个给定的数独（每个给定数独可能有不同的填法），而且要争取更高的总分数。而这个总分数即每个方格上的分值和完成这个数独时填在相应格上的数字的乘积的总和
总分数即每个方格上的分值和完成这个数独时填在相应格上的数字的乘积的总和。如图，在以下的这个已经填完数字的靶形数独游戏中，总分数为 2829。游戏规定，将以总分数的高低决出胜负。
由于求胜心切，小城找到了善于编程的你，让你帮他求出，对于给定的靶形数独，能够得到的最高分数。

输入输出格式

输入格式：

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

输出格式：

输出共 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：

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

输出样例#2：

2852

说明

【数据范围】
40%的数据，数独中非 00 数的个数不少于 3030。
80%的数据，数独中非 00 数的个数不少于2626。
100%的数据，数独中非0 0数的个数不少于 2424。
NOIP 2009 提高组 第四题

Code:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cstdlib>
#include<algorithm>
using namespace std;
long long ans=0;
bool f=0;
int jgg[11][11];
bool h[11][11];
bool l[11][11];
bool g[11][11];
struct cnt {
    int num,tot;
} js[11];
int score[11][11]= {
    {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}
};
void calc() {
    f=1;
    long long sum=0;
    for(int i=1; i<=9; ++i) {
        for(int j=1; j<=9; ++j) {
            sum+=jgg[i][j]*score[i][j];
        }
    }
    if(sum>ans) ans=sum;
}
void dfs(int c,int z,int t) {
    if(jgg[c][z]!=0) {
        if(t==9&&z==9) calc();
        else if(z==9) dfs(js[t+1].num,1,t+1);
        else dfs(c,z+1,t);
    } else {
        for(int i=1; i<=9; ++i) {
            if((!h[c][i])&&(!l[z][i])&&(!g[(c-1)/3*3+(z-1)/3+1][i])) {
                jgg[c][z]=i;
                h[c][i]=1,l[z][i]=1,g[(c-1)/3*3+(z-1)/3+1][i]=1;
                if(t==9&&z==9) calc();
                else if(z==9) dfs(js[t+1].num,1,t+1);
                else dfs(c,z+1,t);
                jgg[c][z]=0;
                h[c][i]=0,l[z][i]=0,g[(c-1)/3*3+(z-1)/3+1][i]=0;
            }
        }
    }
}
bool cmp(cnt a,cnt b) {
    return a.tot<b.tot;
}

int main() {
    for(int i=1; i<=9; ++i)
        js[i].num=i,js[i].tot=0;
    for(int i=1; i<=9; ++i) {
        for(int j=1; j<=9; ++j) {
            scanf("%d",&jgg[i][j]);
            if(jgg[i][j]!=0) {
                if(h[i][jgg[i][j]]==1) {
                    printf("-1\n");
                    return 0;
                }
                if(l[j][jgg[i][j]]==1) {
                    printf("-1\n");
                    return 0;
                }
                if(g[(i-1)/3*3+(j-1)/3+1][jgg[i][j]]==1) {
                    printf("-1\n");
                    return 0;
                }
                h[i][jgg[i][j]]=1,l[j][jgg[i][j]]=1;
                g[(i-1)/3*3+(j-1)/3+1][jgg[i][j]]=1;
            } else js[i].tot++;
        }
    }
    sort(js+1,js+1+9,cmp);
    dfs(js[1].num,1,1);
    if(!f) printf("-1");
    else printf("%lld\n",ans);
    return 0;
}