forked from neetcode-gh/leetcode
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy path52-totalNQueens.js
More file actions
33 lines (27 loc) · 756 Bytes
/
52-totalNQueens.js
File metadata and controls
33 lines (27 loc) · 756 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
var totalNQueens = function (n) {
let col = new Set();
let posDiag = new Set(); // (r + c)
let negDiag = new Set(); // (r - c)
let board = new Array(n).fill().map(() => new Array(n).fill('.'));
let res = 0;
function backtrack(r) {
if (r === n) {
res += 1;
return;
}
for (let c = 0; c < n; c++) {
if (col.has(c) || posDiag.has(r + c) || negDiag.has(r - c)) {
continue;
}
col.add(c);
posDiag.add(r + c);
negDiag.add(r - c);
backtrack(r + 1);
col.delete(c);
posDiag.delete(r + c);
negDiag.delete(r - c);
}
}
backtrack(0);
return res;
};