LeetCode51. N 皇后

题目

分析

依旧回溯的模板题，要看出每次检查一个table，搜索树的元素是table

代码

class Solution {
public:
    vector<vector<string>>res;
    
    bool check(int row,int col,vector<string>table,int n){
        //检查同一列
        for(int i = 0;i < row;i++){
            if(table[i][col] == 'Q')
                return false;
        }
        //检查同一行
        for(int j = 0;j < col;j++){
            if(table[row][j] == 'Q')
                return false;
        }
        //因为回溯是从上到下，从左往右，所以不必检查左下角，右下角
        //检查左上角
        for(int i = row - 1,j = col - 1;i>=0 && j>=0;i--,j--){
            if(table[i][j] == 'Q')
                return false;
        }
        //检查右上角
        for(int i = row -1,j = col + 1;i >=0 && j < n;i--,j++){
            if(table[i][j] == 'Q')
                return false;
        }
        return true;
    }
    void backtracking(int n,int row,vector<string>& table){
        if(row == n){
            res.push_back(table);
            return;
        }
        for(int col = 0;col < n;col++){
            if(check(row,col,table,n)){
                table[row][col] = 'Q';
                backtracking(n,row+1,table);
                table[row][col] = '.';
            }
        }
    }
    vector<vector<string>> solveNQueens(int n) {
        vector<string>table(n,string(n,'.')); //定义一个table
        backtracking(n,0,table);
        return res;
    }
};

注意如何定义一个table，容器还是不熟