LeetCoder_____ N皇后(51)

1. 思路

很经典的深度优先搜索题目了。

• 按照行进行搜索，保证了每行只有一个；
• 用vis1来记录每列是否有旗子，保证每列只有一个
• 左上右下斜线如何记录呢，观察可以发现这条线上的所有坐标x,y的差值是一定的，所以可以用vis2来记录x,y的差值。如果某两个个坐标差值一样表示这两个点在同一斜线上。
• 同理右上左下使用vis3记录x,y两个坐标的和。

2. 代码

class Solution {
public:
    vector<vector<string>> solveNQueens(int n) {
        vector<bool> vis1(n), vis2(2*n), vis3(2*n);
        vector<vector<string>> ret;
        vector<string> temp(n, string(n, '.'));
        dfs(ret, 0, n, vis1, vis2, vis3, temp);
        return ret;
    }

    void dfs(vector<vector<string>>& ret, int cur,int n, vector<bool>& vis1, 
    				vector<bool>& vis2, vector<bool>& vis3, vector<string>& temp)
    {
        if(cur == n)
        {
            ret.push_back(temp);
            return;
        }
        for(int i = 0; i < n; ++i)
        {
            if(vis1[i] || vis2[cur + i] || vis3[cur - i + n]) continue;
            vis1[i] = vis2[cur + i] = vis3[cur - i + n] = true;
            temp[cur][i] = 'Q';
            dfs(ret, cur + 1, n, vis1, vis2, vis3, temp);
            vis1[i] = vis2[cur + i] = vis3[cur - i + n] = false;
            temp[cur][i] = '.';
        }


    }
};