LeetCode 051. N 皇后 DFS
阿新 • • 發佈:2020-09-03
地址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; } voiddfs(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; } };
後面可以考慮映象加速 左右對稱 上下對稱等