N-Queens

最新推荐文章于 2022-06-06 22:03:40 发布

穴工

最新推荐文章于 2022-06-06 22:03:40 发布

阅读量333

点赞数

分类专栏： leetcode 文章标签：简单

本文链接：https://blog.csdn.net/jiongmumu/article/details/9870605

版权

leetcode 专栏收录该内容

53 篇文章 0 订阅

订阅专栏

又是简单深度优先问题

class Solution {
public:

    vector<bool> useCol;
    vector<int> rowNum;
    vector<vector<string>> result;
    
    
    void dfs(vector<string> &board,int n)
    {
        if(board.size()==n)
        {
            result.push_back(board);
            return;
        }
        
        int cur_row=board.size();
        for(int i=0;i<n;i++)
        {
            if(useCol[i]==false)
            {
                bool notConflict=true;
                for(int r=0;r<board.size();r++)
                {
                    if(cur_row-r-i+rowNum[r]==0)
                    {
                        notConflict=false;
                        break;
                    }
                    else if(cur_row-r==rowNum[r]-i)
                    {
                        notConflict=false;
                        break;
                    }
                }
                
                if(notConflict)
                {
                    rowNum[cur_row]=i;
                    board.push_back(dict[i]);
                    useCol[i]=true;
                    dfs(board,n);
                    useCol[i]=false;
                    board.pop_back();
                    
                }
            }
        }
    }
    vector<string> dict;
    vector<vector<string> > solveNQueens(int n) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        useCol.clear();
        useCol.assign(n,false);
        rowNum.resize(n);
        
        result.clear();
        dict.clear();
        
        vector<string> board;
        
        string head="";
        for(int i=0;i<n;i++)
        {
            string tmp(n-1-i,'.');
            dict.push_back(head+'Q'+tmp);
            head+='.';
        }
        
        dfs(board,n);
        return result;
    }
};