leetcode n皇后问题

leetcodeN皇后问题

经典回溯算法例题。

class Solution {
public:
    vector<vector<string>> solveNQueens(int n) {
        vector<vector<string>> solve;//最后返回的结果
        unordered_set<int> column;//因为一个皇后单独占据一列，来记录已经被占用的列
        unordered_set<int> duijiao1;//用来记录已经被占用的对角线，下同
        unordered_set<int> duijiao2;
        vector<int> queens(n,-1);//queens[i],第i行第queens[i]列是皇后
        int row=0;
        search(solve,row,n,column,queens,duijiao1,duijiao2);
        return solve;
    }
    void search(vector<vector<string>> &solve,int &row,int n,unordered_set<int> &column,vector<int> &queens,unordered_set<int> &duijiao1,unordered_set<int> &duijiao2)
    {
        if(row==n)//当row等于n时，说明找到了一种解决方案。
        {
            vector<string> oneMethod=generate(queens,n);//绘图
            solve.push_back(oneMethod);//将当前得到解加需要返回的容器中

        }
        else
        {
            for(int i=0;i<n;i++)
            {
            	//回溯就是枚举每种可能发生的情况，去寻找满足要求的解，因此回溯的复杂度很高
            	//因此，对回溯的搜索树剪枝就变得极为重要(优化)
                if(column.find(i)!=column.end())//如果该列已经放置皇后，就剪掉
                    continue;
                 //剪去两个对角线
                int diagonal1=row-i;
                if(duijiao1.find(diagonal1)!=duijiao1.end())
                    continue;
                int diagonal2=row+i;
                if(duijiao2.find(diagonal2)!=duijiao2.end())
                    continue;
                    //该位置可以放皇后时，将当前的列和对角线加入对应的容器中
                queens[row]=i;
                column.insert(i);
                duijiao1.insert(diagonal1);
                duijiao2.insert(diagonal2);
                int t=row+1;
               //进行下一行
                search(solve,t,n,column,queens,duijiao1,duijiao2);
                //回退
                column.erase(i);
                duijiao1.erase(diagonal1);
                duijiao2.erase(diagonal2);
            }
        }
    }
    vector<string> generate(vector<int> &queens,int n)
    {
        vector<string> oneMethod;
        
        for(int i=0;i<n;i++)
        {
            string s=string(n,'.');
            s[queens[i]]='Q';
            oneMethod.push_back(s);
        }
        return oneMethod;
    }
};