T Day - 5 数独简单版（dfs）

AcWing 1613.数独简单版

考点：DFS DFS应用传送门

之前做过的一个和数独相关的题目 👉 数独检查

数独是一种传统益智游戏，你需要把一个 9×99×9 的数独补充完整，使得图中每行、每列、每个 3×33×3 的九宫格内数字 1∼91∼9 均恰好出现一次。

请编写一个程序填写数独。

输入格式

输入共 99 行，每行包含一个长度为 99 的字符串，用来表示数独矩阵。

其中的每个字符都是 1∼91∼9 或 ..（表示尚未填充）。

输出格式

输出补全后的数独矩阵。

数据保证有唯一解。

输入样例：

.2738..1.
.1...6735
.......29
3.5692.8.
.........
.6.1745.3
64.......
9518...7.
.8..6534.

输出样例：

527389416
819426735
436751829
375692184
194538267
268174593
643217958
951843672
782965341

思路：

从左上角开始枚举每一个空格，然后枚举可以填哪些数字，判断哪些是合法的。（这个过程可以画搜索树来实现）

核心问题：

如何判断当前空格可以填哪些数字？

转化为判定问题：判断当前空格能否填某些数字

方法:开三个bool数组 row[i][j] 表示第i行是否填过数字j ； col[i][j] 表示第i列是否填过数字j；cell [i][j][k] 第i行，第j列的小方块中是否填过数字k。

#include <iostream>
#include <algorithm>
#include <cstring>

using namespace std;

const int N = 10;

char g[N][N];
bool row[N][N],col[N][N],cell[3][3][N];

bool dfs(int x,int y)
{
    if(y == 9) x ++ , y = 0; //一行结束，转到下一行
    if(x == 9) //枚举完毕，输出方案
    {
        for(int i = 0; i < 9; i ++ )
            cout << g[i] << endl;
        return true;
    }
    if(g[x][y] != '.') return dfs(x,y + 1); //不是空格，枚举下一位
    for(int i = 0; i < 9; i ++ ) //否则，枚举这一位能填什么
        if(!row[x][i] && !col[y][i] && !cell[x / 3][y / 3][i]) 
        {
            g[x][y] = i + '1';
            row[x][i] = col[y][i] = cell[x / 3][y / 3][i] = true;
            if(dfs(x,y + 1)) return true; //搜索下一层，下一层有解就return
            row[x][i] = col[y][i] = cell[x / 3][y / 3][i] = false; //恢复现场：保证不同分支不相互影响
            g[x][y] = '.';
        }
        
    return false;
}
int main()
{
    for(int i = 0; i < 9; i ++ ) 
    {
        cin >> g[i];
        for(int j = 0; j < 9; j ++ )
            if(g[i][j] != '.')
            {
                int t = g[i][j] - '1'; //把1~9 映射成 0~8
                row[i][t] = col[j][t] = cell[i / 3][j / 3][t] = true;
            }
    }
    
    dfs(0,0);
    
    return 0;
}