【算法/回溯算法/棋盘问题】题解+详细备注（共2题）

51.N皇后

class Solution {
public:
    vector<vector<string>> result; // 存放符合条件的结果集

    bool isValid(int row,int col,vector<string> &chessboard,int n){
        // 校验列
        for(int i = 0;i<row;++i){
            if(chessboard[i][col] == 'Q') return false;
        }
        // 校验45°
        for(int i = row-1,j = col-1;i>=0&&j>=0;i--,j--){
            if(chessboard[i][j] == 'Q') return false;
        }
        // 校验135°
        for(int i = row-1,j = col+1;i >=0&&j<n;i--,j++){
            if(chessboard[i][j] == 'Q') return false;
        }

        return true;
    }

    void backtracking(int n,int row,vector<string> &chessboard){
    	// 递归终止条件
        if(n == row){
            result.push_back(chessboard);
            return;
        }
		// for循环横向遍历（列）
        for(int i{};i<n;++i){
        	// 传入行、列进行校验
            if(isValid(row,i,chessboard,n)){
                chessboard[row][i] = 'Q';
                backtracking(n,row+1,chessboard);	// 递归纵向遍历（行）
                chessboard[row][i] = '.';	// 回溯
            }
        }
    }

    vector<vector<string>> solveNQueens(int n) {
        vector<string> chessboard(n,string(n,'.'));
        backtracking(n,0,chessboard);

        return result;
    }
};

37.解数独

class Solution {
public:
    bool isValid(int row,int col,char val,vector<vector<char>>& board){
        // 判断行
        for(int i{};i<9;++i){
            if(board[row][i] == val) return false;
        }
        // 判断列
        for(int i{};i<9;++i){
            if(board[i][col] == val) return false;
        }
        // 判断9宫格
       	// 拿到九宫格的左上角坐标
        int startRow = (row/3)*3;
        int startCol = (col/3)*3;
        for(int i{startRow};i<startRow+3;++i){
            for(int j = startCol;j < startCol+3;++j){
                if(board[i][j] == val) return false;
            }
        }

        return true;
    }

    bool backtracking(vector<vector<char>>& board){
        for(int i{};i<9;++i){
            for(int j{};j<9;++j){
            	// 不是空白格、继续
                if(board[i][j] != '.') continue;
                // for循环遍历‘1’-‘9’
                for(char c{'1'} ; c<= '9';++c){
                    if(isValid(i,j,c,board)){
                        board[i][j] = c;
                        if(backtracking(board)) return true; // 递归
                        board[i][j] = '.'; // 回溯
                    }
                }      
                return false;
            }
        }

        return true;
    }

    void solveSudoku(vector<vector<char>>& board) {
        backtracking(board);
    }
};