N皇后问题

---

title: N皇后问题
date: 2019-11-06 20:38:14
tags:

【题目】
  本题leetcode(51.N皇后问题)链接：https://leetcode-cn.com/problems/n-queens/
52题是51题上的简化，不再赘述.
【描述】
   n 皇后问题研究的是如何将 n 个皇后放置在 n×n 的棋盘上，并且使皇后彼此之间不能相互攻击。
【思路】
   (1) 使用递归回溯法，每次填一列，在确定列号的情况下，不断去尝试每一行是否可以放置皇后.
那么第1次dfs的皇后在第0列， (i,0),i在[0,n-1]中取值；
那么第2次dfs的皇后在第1列， (i,1),i在[0,n-1]中取值；
那么第3次dfs的皇后在第2列， (i,2),i在[0,n-1]中取值；
…
…
那么第n次dfs的皇后 (i,n-1),i在[0,n-1]中取值；
那么第n+1次dfs的皇后 (i,n),纵坐标j=n,因为j=n，则结束，并将结果存放到result中；

   (2) 因为皇后有三个约束条件，即行不能相同，列不能相同，不再同一直线上(斜率不为1或-1).
采用4个unordered_map数组进行判断，
行判断，unordered_map<int, bool> rowMap(n); //行
列判断，unordered_map<int, bool> colMap(n); //列
斜率为1判断，unordered_map<int, bool> LineMap1; //斜率为1
斜率为-1判断，unordered_map<int, bool> LineMap2; //斜率为-1

如何判断斜率呢，可以采用记录的方式，比如在填写第Qnum个皇后的时候，可以遍历前Qnum-1个是否和第Qnum个在同一直线上，但这样总共需要额外增加N^2的时间复杂度，则实际上我们是可以用hashmap实现的，查询复杂度为O(1).
斜率为1且在同一直线的坐标，可以发现x+y的值是相等的，例如(2,1)和(1,2)在同一直线上，可以发现2+1=1+2
斜率为-1且在同一直线的坐标，可以发现x-y的值是相等的，例如(3,1)和(4,2)在同一直线上，可以发现3-1=4-2
例如：(3,1)有一个皇后，则LineMap1[3+1] = true;LineMap1[3-1] = true; 进行记录，
当又来一个坐标(4,2)时，发现LineMap1[4-2] = true，则说明该直线上已经有皇后了，则不能填写.

理解清楚这些后，就可以着手进行实现了。

#include <iostream>
#include <vector>
#include <unordered_map>

using namespace std;

//函数声明
void dfs(vector<vector<string>> &result, unordered_map<int, bool> &rowMap,
                      unordered_map<int, bool> &colMap,
                      unordered_map<int, bool> &LineMap1, unordered_map<int, bool> &LineMap2,
                      vector<string> temp_ans, int n, int x, int y, int Qnum);

vector<vector<string>> solveNQueens(int n);


/**
 * @param n  皇后的数量，即方框的大小
 * @return   得到的结果
 */
vector<vector<string>> solveNQueens(int n) {
    unordered_map<int, bool> rowMap(n);   //行
    unordered_map<int, bool> colMap(n);   //列
    unordered_map<int, bool> LineMap1;    //斜率为1
    unordered_map<int, bool> LineMap2;    //斜率为-1

    vector<vector<string>> result;

    string str(n, '.');
    vector<string> temp_ans(n, str);

    dfs_2(result, rowMap, colMap, LineMap1, LineMap2, temp_ans, n, 0, 0, 0);

    return result;
}

/**
 * @param result        保存结果
 * @param rowMap        每一行是否存放元素
 * @param colMap        每一列是否存放元素
 * @param LineMap1      斜率为1的情况，同一条斜率为1的线上(x+y)值相等
 * @param LineMap2      斜率为-1的情况，同一条斜率为-1的线上(x-y)值相等
 * @param temp_ans      临时结果，记录路径
 * @param n             边界大小
 * @param x             x坐标
 * @param y             y坐标
 * @param Qnum          已填的皇后数量
 */
void dfs(vector<vector<string>> &result, unordered_map<int, bool> &rowMap,
                      unordered_map<int, bool> &colMap,
                      unordered_map<int, bool> &LineMap1, unordered_map<int, bool> &LineMap2,
                      vector<string> temp_ans, int n, int x, int y, int Qnum) {
    if (y == n) {
        result.push_back(temp_ans);
    } else {
        for (int i = 0; i < n; i++) { //去尝试第y列的每一行是否可用放元素
            if (rowMap[i] == true) {
                continue;
            }
            if (rowMap[i] == false && LineMap1[i + y] == false &&
                LineMap2[i - y + (n - 1)] == false) { // 检查行、斜率为1和斜率为-1

                LineMap1[i + y] = true;
                LineMap2[i - y + (n - 1)] = true;
                rowMap[i] = true;
                temp_ans[i][y] = 'Q';
                Qnum++;

                dfs(result, rowMap, colMap, LineMap1, LineMap2, temp_ans, n, i, y + 1, Qnum);

                LineMap1[i + y] = false;
                LineMap2[i - y + (n - 1)] = false;
                rowMap[i] = false;
                temp_ans[i][y] = '.';
                Qnum--;
            }
        }
    }
}

int main() {
    vector<vector<string>> result = solveNQueens(4);

    cout << "result.size() = " << result.size() << endl;
    for (int i = 0; i < result.size(); i++) {
        for (int j = 0; j < result[i].size(); j++) {
            cout << result[i][j] << endl;
        }
        cout << endl;
    }
    return 0;
}

运行n = 4,结果如下：

result.size() = 2
..Q.
Q...
...Q
.Q..

.Q..
...Q
Q...
..Q.

即得到两个不同的解法.