数独
数独是一种传统益智游戏,你需要把一个9 × 9的数独补充完整,使得图中每行、每列、每个3 × 3的九宫格内数字1~9均恰好出现一次。
请编写一个程序填写数独。
输入格式
输入包含多组测试用例。
每个测试用例占一行,包含81个字符,代表数独的81个格内数据(顺序总体由上到下,同行由左到右)。
每个字符都是一个数字(1-9)或一个”.”(表示尚未填充)。
您可以假设输入中的每个谜题都只有一个解决方案。
文件结尾处为包含单词“end”的单行,表示输入结束。
输出格式
每个测试用例,输出一行数据,代表填充完全后的数独。
输入样例:
4.....8.5.3..........7......2.....6.....8.4......1.......6.3.7.5..2.....1.4......
......52..8.4......3...9...5.1...6..2..7........3.....6...1..........7.4.......3.
end
输出样例:
417369825632158947958724316825437169791586432346912758289643571573291684164875293
416837529982465371735129468571298643293746185864351297647913852359682714128574936
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
const int N = 9;
char str[100];
// ones 记录当前二进制数有几个 1 map 返回当前二进制数结尾 1 的位置
int ones[1 << N], map[1 << N];
// 记录 col 行 row 列 cell 九宫格 可选项
int col[N], row[N], cell[3][3];
// 返回二进制数中最后一个 1 坐在的位置
inline int lowbit(int x) {
return x & -x;
}
// inline 可以减少调用函数的时间,但对递归函数无效
inline void init() {
// 将每个位置都初始化为可放置数字 1000000000 - 1 = 11111111
for (int i = 0; i < N; i ++) row[i] = col[i] = (1 << N) - 1;
for (int i = 0; i < 3; i ++) {
for (int j = 0; j < 3; j ++) cell[i][j] = (1 << N) - 1;
}
}
// 返回当前格子可选择的数的个数
inline int get(int x, int y) {
return row[x] & col[y] & cell[x / 3][y / 3];
}
bool dfs(int cnt) {
if (!cnt) return true;
int minv = 10;
int x, y;
// 寻找可选择数量最少的格子
for (int i = 0; i < N; i ++) {
for (int j = 0; j < N; j ++) {
if (str[i * N + j] == '.') {
int t = ones[get(i, j)];
if (t < minv) {
minv = t;
x = i, y = j;
}
}
}
}
// 遍历当前可选的状态
for (int i = get(x, y); i; i -= lowbit(i)) {
int t = map[lowbit(i)];
// 填充
// 将当前数代表的状态置 0
row[x] -= 1 << t;
col[y] -= 1 << t;
cell[x / 3][y / 3] -= 1 << t;
// 填充格子
str[x * N + y] = t + '1';
if (dfs(cnt - 1)) return true;
// 恢复现场
// 将当前数代表的状态置 1
row[x] += 1 << t;
col[y] += 1 << t;
cell[x / 3][y / 3] += 1 << t;
// 将格子恢复原状
str[x * N + y] = '.';
}
return false;
}
int main() {
for (int i = 0; i < N; i ++) map[1 << i] = i;
for (int i = 0; i < 1 << N; i ++) {
int s = 0;
// 100010 = 1000 + 10
for (int j = i; j; j -= lowbit(j)) s ++;
ones[i] = s;
}
while (scanf("%s", str), strcmp(str, "end")) {
init();
// 记录需要填充的位置数量
int cnt = 0;
for (int i = 0, k = 0; i < N; i ++) {
for (int j = 0; j < N; j ++, k ++) {
// 存放的是 0 - 8
// 通过已填放数,来更新行 列 九宫格的状态
if (str[i * N + j] != '.') {
int t = str[i * N + j] - '1';
if (t != '.') {
col[j] -= 1 << t;
row[i] -= 1 << t;
cell[i / 3][j / 3] -= 1 << t;
}
}else cnt ++;
}
}
dfs(cnt);
cout << str << endl;
}
return 0;
}