. - 力扣(LeetCode)
设计一种算法,打印 N 皇后在 N × N 棋盘上的各种摆法,其中每个皇后都不同行、不同列,也不在对角线上。这里的“对角线”指的是所有的对角线,不只是平分整个棋盘的那两条对角线。
注意:本题相对原题做了扩展
示例:
输入:4 输出:[[".Q..","...Q","Q...","..Q."],["..Q.","Q...","...Q",".Q.."]] 解释: 4 皇后问题存在如下两个不同的解法。 [[".Q..", // 解法 1"...Q","Q...","..Q."],["..Q.", // 解法 2"Q...","...Q",".Q.."] ]
class Solution {
public:bool isValid(int col, int row, int n, vector<int>& pos) {// 判断当前点是否可以时,只需要判断一下之前处理过的列。for (int c = 1; c < col; c++) {// 如果之前某一列同一行放置过,或者在对角线上if (pos[c] == row || col - c == abs(pos[c] - row)) {return false;}}return true;}void dfs(int col, int n, vector<int>& pos, vector<vector<string>>& res) {for (int r = 1; r <= n; r++) {if (isValid(col, r, n, pos)) {pos[col] = r;if (col == n) {vector<string> all_str(n, string(n, '.'));for (int cc = 1; cc <= n; cc++) {all_str[pos[cc]-1][cc-1] = 'Q';}res.push_back(all_str);break;}dfs(col+1, n, pos, res);// 对于某一列col,针对这一行r,如果已经搜索成功,则需要置为0,这样可以再遍历其他合法的rpos[col] = 0;}}}vector<vector<string>> solveNQueens(int n) {vector<vector<string>> res;vector<int> pos(n+1, 0);// 对于第一个皇后处于第一列,看看有多少中摆法dfs(1, n, pos, res);return res;}
};