LeetCode(37) Sudoku Solver

题目如下:

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

LeetCode(37) Sudoku Solver_第1张图片

LeetCode(37) Sudoku Solver_第2张图片


分析如下:

这道题目基本上和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);
        
    }
};



你可能感兴趣的:(DFS,backtracking)