51. N皇后

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

示例:

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

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

示例:

输入: 4
输出: [
 [".Q..",  // 解法 1
  "...Q",
  "Q...",
  "..Q."],

 ["..Q.",  // 解法 2
  "Q...",
  "...Q",
  ".Q.."]
]
解释: 4 皇后问题存在两个不同的解法。

1.利用递规和剪枝

2.如何快速判断不合法的情况

python 代码

Out[33]: [['.Q..', '...Q', 'Q...', '..Q.'], ['..Q.', 'Q...', '...Q', '.Q..']]

    ...:     def solveNQueens(self, n):
    ...:         """
    ...:         :type n: int
    ...:         :rtype: List[List[str]]
    ...:         """
    ...:         ans=[]
    ...:         def dfs(nums,row):
    ...:             if row==n:
    ...:                 ans.append(nums[:])
    ...:                 return
    ...:             for col in range(n):
    ...:                 nums[row]=col//存放的是每一行queen所在的列
    ...:                 print(row)
    ...:                 if valid(nums,row):
    ...:                     dfs(nums,row+1)
    ...:
    ...:
    ...:         def valid(nums,row):
    ...:             for i in range(row):
    ...:                 if nums[i]==nums[row] or abs(nums[i]-nums[row])==abs(i-row):
    ...:                     return False
    ...:             return True
    ...:
    ...:         dfs([None for _ in range(n)],0)
    //格式转化
    ...:         result=[[] for _ in range(len(ans))]
    ...:         for i in range(len(ans)):
    ...:             for col in ans[i]:
    ...:                 tmp='.'*n
    ...:                 result[i].append(tmp[:col]+'Q'+tmp[col+1:])
    ...:         return result

In [35]: sol=Solution()

In [36]: sol.solveNQueens(4)
0
1
1
1
2
2
2
2
1
2
2
3
3
3
3
2
2
0
1
1
1
1
2
3
3
3
3
2
2
2
0
1
2
2
2
2
3
3
3
3
1
1
1
0
1
2
2
2
3
3
3
3
2
1
2
2
2
2
1
1
Out[36]: [['.Q..', '...Q', 'Q...', '..Q.'], ['..Q.', 'Q...', '...Q', '.Q..']]

In [37]:

c++ 代码

class Solution {
private:
    vector<bool> col, dia1, dia2;
    vector<vector<string>> res;

    // 尝试在一个n皇后问题中, 摆放第index行的皇后位置
    void putQueen(int n, int index, vector<int> &row){

        if(index == n){
            res.push_back(generateBoard(n, row));
            return;
        }

        for(int i = 0 ; i < n ; i ++)
            // 尝试将第index行的皇后摆放在第i列
            if(!col[i] && !dia1[index + i] && !dia2[index - i + n - 1]){
                row.push_back(i);
                col[i] = true;
                dia1[index + i] = true;
                dia2[index - i + n - 1] = true;
                putQueen(n, index + 1, row);
                col[i] = false;
                dia1[index + i] = false;
                dia2[index - i + n - 1] = false;
                row.pop_back();
            }

        return;
    }

    vector<string> generateBoard(int n, vector<int> &row){
        assert(row.size() == n);
        vector<string> board(n, string(n, '.'));
        for(int i = 0 ; i < n ; i ++)
            board[i][row[i]] = 'Q';
        return board;
    }

public:
    vector<vector<string>> solveNQueens(int n) {

        res.clear();

        col.clear();
        for(int i = 0 ; i < n ; i ++)
            col.push_back(false);

        dia1.clear();
        dia2.clear();
        for(int i = 0 ; i < 2 * n - 1 ; i ++){
            dia1.push_back(false);
            dia2.push_back(false);
        }

        vector<int> row;
        putQueen(n, 0, row);

        return res;
    }
};


// void printBoard(const vector<string>& board){
//     for(string row: board)
//         cout << row << endl;
//     cout << endl;
// }

// int main() {

//     int n = 4;
//     vector<vector<string>> res = Solution().solveNQueens(n);
//     for(vector<string> board: res)
//         printBoard(board);

//     return 0;
// }

参考：https://www.cnblogs.com/ariel-dreamland/p/9143352.html