Leetcode 51. N-Queens

51. N-Queens

The n-queens puzzle is the problem of placing n queens on an n×n chessboard such that no two queens attack each other.

Given an integer n, return all distinct solutions to the n-queens puzzle.

Each solution contains a distinct board configuration of the n-queens’ placement, where ‘Q’ and ‘.’ both indicate a queen and an empty space respectively.

Example:

Input: 4
Output:

[
 [".Q..",  // Solution 1
  "...Q",
  "Q...",
  "..Q."],

 ["..Q.",  // Solution 2
  "Q...",
  "...Q",
  ".Q.."]
]

Explanation: There exist two distinct solutions to the 4-queens puzzle as shown above.

思路： 皇后之间不能互相攻击的前提是，皇后不能同在一条直线上（包括横竖两个方向）和皇后不能再同一对角线上。

根据这个规则，每一行只能摆放一个皇后，对这个皇后尝试所有可能的列（从第1-第n列，n为棋盘大小），从第一行开始，对每一行皇后的位置尝试所有可能，逐行开始摆放皇后，如果遇到某一行皇后在任何位置摆放都会与之前摆放的皇后冲突，那么回溯到前一步，修改上一行皇后的位置，若还不行则依次回溯。

代码

class Solution {
public:
    vector<vector<string>> ans;//存放最后的答案
    
    vector<vector<string>> solveNQueens(int n) {
        vector<int> position(n+1, 0);//存放皇后摆放的位置，第i个位置上的数值表示第i行皇后所在的列
        if (n!=0)
            NQueens(position, n, 1);//从棋盘第一行开始尝试摆放皇后
        return ans;
    }
    
    void NQueens(vector<int>& position, int n, int k)//n个皇后需要摆放，假设前k-1行皇后已经摆放妥当，该函数尝试摆放第k行的皇后
    {
        if(k == n+1)
        {
            vector<string> rst(n, string(n, '.'));
            for (int i = 1; i<=n; i++)
            {
                rst[i-1][position[i]-1] = 'Q';
            }
            ans.push_back(rst);
            return;
        }
        int i, j;
        for (i = 1; i <= n; i++)//第k行尝试所有位置[1-n]
        {
            for(j = 1; j<k; j++)//遍历position，找到第k行皇后的合适位置
            {
                if (position[j]==i || abs(j-k)==abs(position[j]-i))//第k行尝试的位置与之前已经放好的皇后冲突
                    break;//尝试下一种摆放方法
            }
            if (j == k)//上面的for循环正常结束，没有冲突
            {
                position[k] = i;//摆放第k行的皇后
                NQueens(position, n, k+1);
            }
        }
    }
};