题目如下:
Write a program to solve a Sudoku puzzle by filling the empty cells.
Empty cells are indicated by the character '.'.
You may assume that there will be only one unique solution
分析如下:
这道题目基本上和N-QueenI,N-Queen II 是一个套路,套用DFS + Backtracking的思路就可以,代码也是一个风格的。
不过这两题有个区别,需要注意一下。
在N-Queen I 中,返回所有正确的解,N-Queen II中,返回所有正确的解的数量,无论是N-Queen I 还是N-Queen II,都需要找到所有的解。
换句话说,需要把search tree全部搜索一遍,才能得到答案,如果只搜到某一个答案就返回,得到的结果是错误的。
在Sudoku中,只需要把找到的第一个解(并且题目的输入保证了将只产生一个解)返回就可以了,如果不立刻返回而是继续在搜索树(搜索树的定义见这里的图1,图2)中遍历的话,将得到错误的答案。
Version 1:
//77ms
class Solution {
private:
bool isOver; //是否已经找到一个正确解。
public:
void dfs (vector<vector<char> > &board, int i, int j, int n ) {
//if (this->isOver) return;
if (j >= n) {
dfs(board, i+1, 0, n);
} else if (i == n){
this->isOver = true;
return ;
} else if (board[i][j] != '.'){
dfs(board, i, j+1, n);
} else {
for (int k = 1; k <= n; ++k){
board[i][j] = (char)('0' + k);
if (isValid(board, i, j, n)){
dfs(board, i, j+1 , n);
}
if (this->isOver) return; //找到一个解救跳出,防止遍历整个search tree
board[i][j] = '.';
}
}
return ;
}
bool isValid(vector<vector<char>> &board, int i , int j, int n) {
for (int index = 0; index < n; ++index){
if (index != j && board[i][index] == board[i][j]) {
return false;
}
}
for (int index = 0; index < n; ++index){
if (index != i && board[index][j] == board[i][j]) {
return false;
}
}
int index_i = i / 3;
int index_j = j / 3;
for (int x = index_i * 3; x < index_i * 3 + 3; ++x) {
for (int y = index_j * 3; y < index_j * 3 + 3; ++y) {
if ((x!=i || y != j) && board[x][y] == board[i][j]) { //注意这里逻辑是"或"||,不是"与"
return false;
}
}
}
return true;
}
void solveSudoku(vector<vector<char> > &board) {
if (board.size() == 0 || board[0].size() == 0) return; //简洁的方式去验证输入合法性
this->isOver = false;
dfs(board, 0, 0, 9);
}
};
Version 2:
// 69ms
class Solution {
public:
bool dfs (vector<vector<char> > &board, int i, int j, int n ) { //函数写成return bool型,这样也能帮助在找到第一个解的时候return.
if (j >= n) {
return dfs(board, i+1, 0, n);
} else if (i == n){
return true;
} else if (board[i][j] != '.'){
return dfs(board, i, j+1, n);
} else {
for (int k = 1; k <= n; ++k){
board[i][j] = (char)('0' + k);
if (isValid(board, i, j, n)){
if (dfs(board, i, j+1 , n)) // 如果找到第一个解,就及时return
return true;
}
board[i][j] = '.';
}
}
return false;
}
bool isValid(vector<vector<char>> &board, int i , int j, int n) {
for (int index = 0; index < n; ++index){
if (index != j && board[i][index] == board[i][j]) {
return false;
}
}
for (int index = 0; index < n; ++index){
if (index != i && board[index][j] == board[i][j]) {
return false;
}
}
int index_i = i / 3;
int index_j = j / 3;
for (int x = index_i * 3; x < index_i * 3 + 3; ++x) {
for (int y = index_j * 3; y < index_j * 3 + 3; ++y) {
if ((x!=i || y != j) && board[x][y] == board[i][j]) { //注意这里逻辑是"或"||,不是"与"
return false;
}
}
}
return true;
}
void solveSudoku(vector<vector<char> > &board) {
if (board.size() == 0 || board[0].size() == 0) return; //简洁的方式去验证输入合法性
dfs(board, 0, 0, 9);
}
};