leetcode51

N皇后

N皇后问题要求同一行、同一列还有斜对角线不能同时出现两个及以上的皇后。

思路：满足题目的三个需求，需要对棋盘的深度进行递归，递归的过程中进行回溯，也就是撤销上一次的移动。

class Solution {
private:
    vector<vector<string>> ans;

    void slove(int n,int r, vector<string> &temp){
        if(r == n){
            ans.push_back(temp);
            return;
        }
        for(int c = 0;c<n;c++){
            if(isValid(n,r,c,temp)){
                temp[r][c] = 'Q';
                slove(n,r+1,temp);
                    //回溯
                temp[r][c] = '.';
            }
        }
    }

    bool isValid(int n,int r,int c,vector<string> &path){
       //检查同一列
        for(int i =0;i<r;i++){
            if(path[i][c] =='Q')
                return false;
        }

        //判断两条对角线
        for(int i = r-1,j=c-1;i>=0&&j>=0; i--,j--){
            if(path[i][j] == 'Q')
                return false;
        }

        for(int i =r-1,j=c+1;i>=0&&j<n;i--,j++){
            if(path[i][j] == 'Q')
                return false;
        }

        return true;
    }
public:
    vector<vector<string>> solveNQueens(int n) {
        ans.clear();
        vector<string> temp(n, string(n, '.'));
        slove(n,0,temp);
        return ans;
    }
};

看完之后，你是不是发现了代码中没有对棋盘的同一行进行判断，这是因为每一层递归只会选for循环 当前行 的一个元素。