LeetCode 51.N皇后
原题链接:N皇后
DFS回溯。注意JavaScript字符串是不可变对象,plan
用二维字符数组表示以便修改。
JavaScript
/**
* @param {number} n
* @return {string[][]}
*/
var solveNQueens = function (n) {
const visn = Array(n).fill(false);//列
const visne = Array(n * 2).fill(false);//正对角线
const visnw = Array(n * 2).fill(false);//反对角线
const ans = [], plan = [];
for (let i = 0; i < n; i++) plan.push(Array(n).fill("."));
function dfs(r) {
if (r == n) {
ans.push(plan.map(i => i.join("")));
return;
}
for (let c = 0; c < n; c++) {
if (!visn[c] && !visne[r + c] && !visnw[n + r - c]) {
visn[c] = visne[r + c] = visnw[n + r - c] = true;
plan[r][c] = "Q";
dfs(r + 1);
visn[c] = visne[r + c] = visnw[n + r - c] = false;
plan[r][c] = ".";
}
}
}
dfs(0);
return ans;
};