1. 程式人生 > 實用技巧 >LeetCode 051. N 皇后 DFS

LeetCode 051. N 皇后 DFS

地址https://leetcode-cn.com/problems/n-queens/

n 皇后問題研究的是如何將 n 個皇后放置在 n×n 的棋盤上,
並且使皇后彼此之間不能相互攻擊。
給定一個整數 n,
返回所有不同的n皇后問題的解決方案。

每一種解法包含一個明確的n 皇后問題的棋子放置方案,
該方案中 'Q''.' 分別代表了皇后和空位。



示例:

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

 ["..Q.",  // 解法 2
  "Q...",
  "...Q",
  ".Q.."]
]
解釋: 
4 皇后問題存在兩個不同的解法。

上圖為 8 皇后問題的一種解法。

演算法1
經典DFS 每行嘗試防止一個皇后 檢測皇后上下左右斜 八個方向是否有衝突

class Solution {
public:
 vector<vector<string>> ans;

int addx[8] = { 1,-1,1,-1,0,0,1,-1};
int addy[8] = { 1,-1,-1,1,1,-1,0,0 };

bool Check(const vector<string>& v, int x, int y)
{
    int n = v.size();

    
for (int i = 0; i < 8; i++) { int xx = x+addx[i]; int yy = y+addy[i]; while (xx >= 0 && xx < n && yy >= 0 && yy < n) { if (v[xx][yy] == 'Q') return false; xx += addx[i]; yy += addy[i]; } } return true; } void
dfs(vector<string>& v, int x) { if (x == v.size() ) { ans.push_back(v); return; } for (int y = 0; y < v.size(); y++) { v[x][y] = 'Q'; if(Check(v,x,y)) dfs(v,x + 1); v[x][y] = '.'; } return; } vector<vector<string>> solveNQueens(int n) { vector<string> v(n, string(n, '.')); dfs(v, 0); return ans; } };

後面可以考慮映象加速 左右對稱 上下對稱等