【题目】**37. 解数独
编写一个程序,通过已填充的空格来解决数独问题。
一个数独的解法需遵循如下规则:
数字 1-9 在每一行只能出现一次。
数字 1-9 在每一列只能出现一次。
数字 1-9 在每一个以粗实线分隔的 3x3 宫内只能出现一次。
空白格用 ‘.’ 表示。
一个数独。
答案被标成红色。
Note:
给定的数独序列只包含数字 1-9 和字符 ‘.’ 。
你可以假设给定的数独只有唯一解。
给定数独永远是 9x9 形式的。
【解题思路1】递归回溯
class Solution {
private boolean[][] line = new boolean[9][9];
private boolean[][] column = new boolean[9][9];
private boolean[][][] block = new boolean[3][3][9];
private boolean valid = false;
private List<int[]> spaces = new ArrayList<int[]>();
public void solveSudoku(char[][] board) {
// 将所有的空格找出来,并将已有的数组做上标记
for (int i = 0; i < 9; ++i) {
for (int j = 0; j < 9; ++j) {
if (board[i][j] == '.') {
spaces.add(new int[]{i, j});
} else {
int digit = board[i][j] - '0' - 1;
line[i][digit] = column[j][digit] = block[i / 3][j / 3][digit] = true;
}
}
}
dfs(board, 0);
}
public void dfs(char[][] board, int pos) {
// 填满了空格
if (pos == spaces.size()) {
valid = true;
return;
}
int[] space = spaces.get(pos);
int i = space[0], j = space[1];
for (int digit = 0; digit < 9 && !valid; ++digit) {
if (!line[i][digit] && !column[j][digit] && !block[i / 3][j / 3][digit]) {
line[i][digit] = column[j][digit] = block[i / 3][j / 3][digit] = true;
board[i][j] = (char) (digit + '0' + 1);
dfs(board, pos + 1);
line[i][digit] = column[j][digit] = block[i / 3][j / 3][digit] = false; // 回溯
}
}
}
}
【解题思路2】位运算优化(待研究)
整体的递归 + 回溯的框架与上面是一致的,仅仅将一个数组「压缩」成了一个数,仅使用一个整数表示每个数字是否出现过。
具体地,数 b 的二进制表示的第 i 位(从低到高,最低位为第 0 位)为 1,当且仅当数字 i+1 已经出现过。例如当 b 的二进制表示为 (011000100)2 时,就表示数字 3,7,8 已经出现过。
位运算有一些基础的使用技巧。下面列举了所有在代码中使用到的技巧: (这里哭了,位运算还是考验人啊)
- 对于第 i 行第 j 列的位置,line[i] ∣ column[j] ∣ block[⌊i/3⌋][⌊j/3⌋]中第 k 位为 1,表示该位置不能填入数字 k+1(因为已经出现过),其中 ∣ 表示按位或运算。如果我们对这个值进行 ∼ 按位取反运算,那么第 k 位为 1 就表示该位置可以填入数字 k+1,我们就可以通过寻找 1 来进行枚举。由于在进行按位取反运算后,这个数的高位也全部变成了 1,而这是我们不应当枚举到的,因此我们需要将这个数和 ( 111111111 ) 2 = ( 1FF ) 16 ( 111111111 ) 2 = ( 1 F F ) 16 (111111111)_2 = (\text{1FF})_{16}(111111111)2=(1FF)16 (111111111)2=(1FF)16(111111111)2=(1FF)16 进行按位与运算 &,将所有无关的位置为 0;
- 可以使用按位异或运算 ∧,将第 i 位从 0 变为 1,或从 1 变为 0。具体地,与数 1<<i 进行按位异或运算即可,其中 << 表示左移运算;
- 可以用 b & (−b) 得到 b 二进制表示中最低位的 1,这是因为 (−b) 在计算机中以补码的形式存储,它等于 ∼b+1。b 如果和 ∼b 进行按位与运算,那么会得到 0,但是当 ∼b 增加 1 之后,最低位的连续的 1 都变为 0,而最低位的 0 变为 1,对应到 b 中即为最低位的 1,因此当 b 和 ∼b+1 进行按位与运算时,只有最低位的 1 会被保留;
- 当得到这个最低位的 1 时,可以通过一些语言自带的函数得到这个最低位的 1 究竟是第几位(即 i 值),具体可以参考下面的代码;
- 可以用 b 和最低位的 1 进行按位异或运算,就可以将其从 b 中去除,这样就可以枚举下一个 1。同样地,我们也可以用 b 和 b−1 进行按位与运算达到相同的效果,读者可以自行尝试推导。
class Solution {
private int[] line = new int[9];
private int[] column = new int[9];
private int[][] block = new int[3][3];
private boolean valid = false;
private List<int[]> spaces = new ArrayList<int[]>();
public void solveSudoku(char[][] board) {
for (int i = 0; i < 9; ++i) {
for (int j = 0; j < 9; ++j) {
if (board[i][j] == '.') {
spaces.add(new int[]{i, j});
} else {
int digit = board[i][j] - '0' - 1;
flip(i, j, digit);
}
}
}
dfs(board, 0);
}
public void dfs(char[][] board, int pos) {
if (pos == spaces.size()) {
valid = true;
return;
}
int[] space = spaces.get(pos);
int i = space[0], j = space[1];
int mask = ~(line[i] | column[j] | block[i / 3][j / 3]) & 0x1ff;
for (; mask != 0 && !valid; mask &= (mask - 1)) {
int digitMask = mask & (-mask);
int digit = Integer.bitCount(digitMask - 1);
flip(i, j, digit);
board[i][j] = (char) (digit + '0' + 1);
dfs(board, pos + 1);
flip(i, j, digit);
}
}
public void flip(int i, int j, int digit) {
line[i] ^= (1 << digit);
column[j] ^= (1 << digit);
block[i / 3][j / 3] ^= (1 << digit);
}
}
还能继续优化
如果一个空白格只有唯一的数可以填入,也就是其对应的 b 值和 b−1 进行按位与运算后得到 0(即 b 中只有一个二进制位为 1),就可以确定这个空白格填入的数,而不用等到递归时再去处理它。
这样一来,我们可以不断地对整个数独进行遍历,将可以唯一确定的空白格全部填入对应的数。随后再使用与方法二相同的方法对剩余无法唯一确定的空白格进行递归 + 回溯。
class Solution {
private int[] line = new int[9];
private int[] column = new int[9];
private int[][] block = new int[3][3];
private boolean valid = false;
private List<int[]> spaces = new ArrayList<int[]>();
public void solveSudoku(char[][] board) {
for (int i = 0; i < 9; ++i) {
for (int j = 0; j < 9; ++j) {
if (board[i][j] != '.') {
int digit = board[i][j] - '0' - 1;
flip(i, j, digit);
}
}
}
while (true) {
boolean modified = false;
for (int i = 0; i < 9; ++i) {
for (int j = 0; j < 9; ++j) {
if (board[i][j] == '.') {
int mask = ~(line[i] | column[j] | block[i / 3][j / 3]) & 0x1ff;
if ((mask & (mask - 1)) == 0) {
int digit = Integer.bitCount(mask - 1);
flip(i, j, digit);
board[i][j] = (char) (digit + '0' + 1);
modified = true;
}
}
}
}
if (!modified) {
break;
}
}
for (int i = 0; i < 9; ++i) {
for (int j = 0; j < 9; ++j) {
if (board[i][j] == '.') {
spaces.add(new int[]{i, j});
}
}
}
dfs(board, 0);
}
public void dfs(char[][] board, int pos) {
if (pos == spaces.size()) {
valid = true;
return;
}
int[] space = spaces.get(pos);
int i = space[0], j = space[1];
int mask = ~(line[i] | column[j] | block[i / 3][j / 3]) & 0x1ff;
for (; mask != 0 && !valid; mask &= (mask - 1)) {
int digitMask = mask & (-mask);
int digit = Integer.bitCount(digitMask - 1);
flip(i, j, digit);
board[i][j] = (char) (digit + '0' + 1);
dfs(board, pos + 1);
flip(i, j, digit);
}
}
public void flip(int i, int j, int digit) {
line[i] ^= (1 << digit);
column[j] ^= (1 << digit);
block[i / 3][j / 3] ^= (1 << digit);
}
}