代码随想录 第七章 回溯算法 part04 37.解数独

37.解数独

class Solution {
private:
    bool isValid(vector<vector<char>>& board, int x, int y){
        vector<int>ones(10,1);
        vector<int>ct=ones;
        for(int i=0;i<9;i++){
            if(board[x][i]!='.'){
                if(--ct[board[x][i]-'0']<0) return false;
            }
        }
        ct=ones;
        for(int i=0;i<9;i++){
            if(board[i][y]!='.'){
                if(--ct[board[i][y]-'0']<0) return false;
            }
        }
        ct=ones;
        int x9=(x/3)*3,y9=(y/3)*3;
        for(int i=0;i<3;i++){
            for(int j=0;j<3;j++){
                if(board[x9+i][y9+j]!='.'){
                    if(--ct[board[x9+i][y9+j]-'0']<0) return false;
                }
            }
        }
        return true;
    }
    bool bt(vector<vector<char>>& broad){
        for(int i=0;i<9;i++){
            for(int j=0;j<9;j++){
                if(broad[i][j]!='.') continue;
                for(char num='1';num<='9';num++){
                    broad[i][j]=num;
                    if(isValid(broad,i,j)){
                        if(bt(broad)) return true;
                    }
                    broad[i][j]='.';
                }
                return false;
            }
        }
        return true;
    }
public:
    void solveSudoku(vector<vector<char>>& board) {
        bt(board);
    }
};

这题笔者一开始设想的回溯函数需要的参数包括当前数独数组与要进行回溯的元素坐标，但是在编写过程中发现，这样写对于如何回溯与合适结束回溯需要比较复杂的判断，所以在这里还是使用了随想录所给的思路，即不传递下一格需要回溯的元素位置，而是由回溯函数自行寻找仍然没填上数字的位置，这样虽然会增加一定的时长，但是对于何时结束回溯的判断逻辑比较简明。