forked from neetcode-gh/leetcode
-
Notifications
You must be signed in to change notification settings - Fork 0
/
0051-n-queens.cs
51 lines (42 loc) · 1.83 KB
/
0051-n-queens.cs
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
public class Solution {
public IList<IList<string>> SolveNQueens(int n)
{
var result = new List<IList<string>>();
IList<StringBuilder> board = new List<StringBuilder>();
for (int i = 0; i < n; i++)
{
board.Add(new StringBuilder(n));
board[i].Append('.', n);
}
backtrackingNQueens(n, 0, board, result, new HashSet<(int i, int j)>(), new HashSet<(int i, int j)>(), new HashSet<(int i, int j)>());
return result;
}
private void backtrackingNQueens(int n, int row, IList<StringBuilder> board, List<IList<string>> result, HashSet<(int i, int j)> col, HashSet<(int i, int j)> positiveDiag, HashSet<(int i, int j)> negativeDiag)
{
if (n == 0)
{
result.Add(board.Select(s => s.ToString()).ToList());
return;
}
if (row == board.Count) return;
for (int c = 0; c < board.Count; c++)
{
(int i, int j) column = (0, c);
int m = Math.Min(row, c);
(int i, int j) diag1 = (row - m, c - m);
m = Math.Min(row, board.Count - 1 - c);
(int i, int j) diag2 = (row - m, c + m);
if (col.Contains(column) || positiveDiag.Contains(diag1) ||
negativeDiag.Contains(diag2)) continue;
col.Add(column);
positiveDiag.Add(diag1);
negativeDiag.Add(diag2);
board[row][c] = 'Q';
backtrackingNQueens(n - 1, row + 1, board, result, col, positiveDiag, negativeDiag);
board[row][c] = '.';
col.Remove(column);
positiveDiag.Remove(diag1);
negativeDiag.Remove(diag2);
}
}
}