[leetcode刷题系列]N-Queens

最新推荐文章于 2019-06-30 12:57:14 发布

sigh1988

最新推荐文章于 2019-06-30 12:57:14 发布

阅读量563

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/sigh1988/article/details/9738133

版权

leetcode 专栏收录该内容

127 篇文章 0 订阅

订阅专栏

经典题目了，递归搜索就行了

class Solution {
    const static int MAXN = 100;
    bool col[MAXN], add[MAXN], minus[MAXN];
    int ans[MAXN];
    void dfs(int now, int n, vector<vector<string> >& ret){
        if(now >= n){
            vector<string> vs;
            for(int i = 0; i < n; ++ i){
                string s = "";
                for(int j = 0; j < n; ++ j)
                    if(j == ans[i])
                        s += 'Q';
                    else
                        s += '.';
                vs.push_back(s);
            }
            ret.push_back(vs);
            return ;
        }
        for(int i = 0; i < n; ++ i){
            if(!col[i] && !add[now + i] && !minus[now - i + n]){
                col[i] = true;
                add[now + i] = true;
                minus[now - i + n] = true;
                ans[now] = i;
                dfs(now + 1, n, ret);
                col[i] = false;
                add[now + i] = false;
                minus[now - i + n] = false;
            }
        }
    }
public:
    vector<vector<string> > solveNQueens(int n) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        memset(col, 0x00, sizeof(col));
        memset(add, 0x00, sizeof(add));
        memset(minus, 0x00, sizeof(minus));
        vector<vector<string> > ret;
        dfs(0, n, ret);
        return ret;
    }
};