703. 数独检查 - AcWing题库

https://www.acwing.com/problem/content/705/
数独是一种流行的单人游戏。

目标是用数字填充9x9矩阵，使每列，每行和所有9个非重叠的3x3子矩阵包含从1到9的所有数字。

每个9x9矩阵在游戏开始时都会有部分数字已经给出，通常有一个独特的解决方案。

1.png

2.png

给定完成的N2∗N2数独矩阵，你的任务是确定它是否是有效的解决方案。

有效的解决方案必须满足以下条件：

每行包含从1到N2的每个数字，每个数字一次。
每列包含从1到N2的每个数字，每个数字一次。
将N2∗N2矩阵划分为N2个非重叠N∗N子矩阵。 每个子矩阵包含从1到N2的每个数字，每个数字一次。
你无需担心问题的唯一性，只需检查给定矩阵是否是有效的解决方案即可。

输入格式
第一行包含整数T，表示共有T组测试数据。

每组数据第一行包含整数N。

接下来N2行，每行包含N2个数字（均不超过1000），用来描述完整的数独矩阵。

输出格式
每组数据输出一个结果，每个结果占一行。

结果表示为“Case #x: y”，其中x是组别编号（从1开始），如果给定矩阵是有效方案则y是Yes，否则y是No。

数据范围
1≤T≤100,
3≤N≤6

#include<cstdio>
#include<cmath>
#include<algorithm>
#include<string>
#include<iostream>
#include<cstring>
#include<vector>
using namespace std;
#define  ll long long
#define N 40
int a[N][N];
int v[N];
int m,n;
bool check_row()
{
    for(int i=0; i<m; ++i){
        memset(v,0,sizeof(v));
        for(int j=0; j<m; ++j){
            if(a[i][j]>m || a[i][j]<1) return false;
            if(!v[a[i][j]]){
                v[a[i][j]] = 1;
            }
            else{
                return false;
            }
        }
    }
    return true;
}
bool check_col()
{
    for(int i=0; i<m; ++i){
        memset(v,0,sizeof(v));
        for(int j=0; j<m; ++j){
            int t = a[j][i];
            if(t>m || t<1) return false;
            if(!v[t]){
                v[t] = 1;
            }
            else{
                return false;
            }
        }
    }
    return true;
}
bool check_k()//检查块
{
    for(int i=0; i<m; i+=n){
        for(int j=0; j<m; j+=n){
            memset(v,0,sizeof(v));
            for(int dx=0; dx<n; ++dx){  //对于最左上角的x轴偏移量
                for(int dy=0; dy<n; ++dy){//对于最左上角的y轴偏移量
                    int xx = i+dx;
                    int yy = j+dy;
                    int t = a[xx][yy];
                    if(t>m || t<1) return false;
                    if(!v[t]){
                        v[t] = 1;
                    }
                    else{
                        return false;
                    }
                }
            }
        }
    }
    return true;
}
int main(){
    int t;
    cin>>t;
    for(int xx=1; xx<=t; ++xx){
        cin>>n;
        m = n*n;
        for(int i=0; i<m; ++i)
            for(int j=0; j<m; ++j)
                cin>>a[i][j];
        if(check_row() && check_col() && check_k()){
            printf("Case #%d: Yes\n",xx);
        }
        else{
            printf("Case #%d: No\n",xx);
        }
    }
    return 0;
}