LeetCode每日一题 - N 皇后

题目：N皇后

Description

n 皇后问题研究的是如何将 n 个皇后放置在 n×n 的棋盘上，并且使皇后彼此之间不能相互攻击。

上图为 8 皇后问题的一种解法。

给定一个整数 n，返回所有不同的 n 皇后问题的解决方案。

每一种解法包含一个明确的 n 皇后问题的棋子放置方案，该方案中 ‘Q’ 和 ‘.’ 分别代表了皇后和空位。

Sample

输入：4
输出：[
[".Q…", // 解法 1
“…Q”,
“Q…”,
“…Q.”],
["…Q.", // 解法 2
“Q…”,
“…Q”,
“.Q…”]
]
解释: 4 皇后问题存在两个不同的解法。

皇后彼此不能相互攻击，也就是说：任何两个皇后都不能处于同一条横行、纵行或斜线上。

Solution

经典问题！不断试探、纠正！！！
利用布尔数组判断是否会发生冲突，如果已经有皇后放在了那里，设为1，否则设为0。
判断对角线冲突时，有个比较好的技巧：
由于每条主对角线的（x-y）是一定的，每条副对角线的（x+y）也是一定的。
所以可以用（x-y）表示主对角线，用（x+y）表示副对角线。跟处理列的情况统一了！！！
因为（x-y）可能会出现负数，所以可以用（x-y+n）表示主对角线！都是作为数组下标表示！

AC Code

class Solution {
public:
    int n;
    int a[20];
    bool vis[20];//列是否被占
    bool z[20];//主对角线是否被占
    bool f[20];//副对角线是否被占
    vector<vector<string> > v;
    void write() {//存解法
        vector<string> temp;
        for (int i = 1; i <= n; i++) {
            string s;
            for (int j = 1; j <= n; j++) {
                if (j == a[i]) s+='Q';
                else s+='.';
            }
            temp.push_back(s);
        }
        v.push_back(temp);
    }
    void dfs(int dep) {//dep表示当前行数
        if (dep > n) {//找到一个解法
            write(); return;
        }
        for (int i = 1; i <= n; i++) {
            if (!vis[i] && !z[i + dep] && !f[i - dep + n]) {//相应的第i列，主对角线，副对角线都没有被占
                vis[i] = 1; z[i + dep] = 1; f[i - dep + n] = 1;//占领
                a[dep] = i;//将皇后放在了第dep行第i列
                dfs(dep + 1);
                vis[i] = 0; z[i + dep] = 0; f[i - dep + n] = 0;//回溯
            }
        }
    }
    vector<vector<string>> solveNQueens(int m) {
        n=m;
        dfs(1);
        return v;
    }
};

PS：希望对你有帮助！