LeetCode: N-Queens

最新推荐文章于 2019-04-09 23:32:41 发布

原创最新推荐文章于 2019-04-09 23:32:41 发布 · 500 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

144 篇文章

订阅专栏

本文介绍了一种使用深度优先搜索解决N皇后问题的方法。通过递归地在棋盘上放置皇后，并检查每一步是否会产生冲突，最终找到所有可行的解决方案。代码实现包括冲突检测和搜索过程。

思路：经典的深度搜索题目，当前行尝试将Queen放在每列，如果和前面放好的Queen没有冲突，则开始下一行的搜索，否则，放到下一列，进行同样的冲突检测操作。

code:

class Solution {
public:
    bool isAvailable(vector<string> &board, int r, int c,int n){//与放好的Queen进行冲突检测
        for(int i = 0;i<r;i++)
            if(board[i][c] == 'Q')return false;
        for(int j = 0;j<c;j++)
            if(board[r][j] == 'Q')return false;
        int i = r-1, j = c-1;
        while(i>=0 && j >= 0){
            if(board[i][j] == 'Q')return false;
            i--;
            j--;
        }
        i = r - 1;
        j = c + 1;
        while(i>=0 && j<n){
            if(board[i][j] == 'Q')return false;
            i--;
            j++;            
        }
        return true;
    }
    void NQuees(vector<string> &curRet, int r, int n, vector<vector<string> > &ret){
        if(r >= n){//棋盘中每一行都放了一个Queen，存好结果
            ret.push_back(curRet);
            return;
        }
        string s(n,'.');
        curRet.push_back(s);
        for(int j = 0;j<n;j++){
            vector<string> temp = curRet;
            if(isAvailable(curRet,r,j,n)){
                temp[r][j] = 'Q';//没有冲突，在(r,j)位置放置一个Queen
                NQuees(temp,r+1,n,ret);
            }
            else
                temp[r][j] = '.';//有冲突，不能放置Queen
        }
    }
    vector<vector<string> > solveNQueens(int n) {
        vector<string> board;
        vector<vector<string> > ret;
        NQuees(board,0,n,ret);
        return ret;
    }
};