Leecode 面试题 08.12. 八皇后

原题链接：Leecode 面试题 08.12. 八皇后

class Solution {
public:
    vector<vector<string>> res;
    bool v[9];
    int col[9];
    void dfs(int n,int now)
    {
        if(n==now)
        {
            vector<string> v;
            for(int i=0;i<n;i++)
            {
                string s;
                for(int j=0;j<n;j++)
                {
                    if(col[j]!=i) s+='.';
                    else s+='Q';
                }
                v.push_back(s);
            }
            res.push_back(v);
        }
        for(int i=0;i<n;i++)
        {
            if(!v[i])
            {
                bool flag=true;
                for(int j=0;j<now;j++)
                {
                    if(abs(i-col[j])==abs(j-now))
                    {
                        flag=false;
                        break;
                    }
                }
                if(flag)
                {
                    v[i]=true;
                    col[now]=i;
                    dfs(n,now+1);
                    v[i]=false;
                }
            }
        }
    }
    vector<vector<string>> solveNQueens(int n) {
        dfs(n,0);
        return res;
    }
};