forked from striver79/SDESheet
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathNQueenBrute
60 lines (52 loc) · 1.41 KB
/
NQueenBrute
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
52
53
54
55
56
57
58
59
60
class Solution {
public:
bool isSafe1(int row, int col, vector<string> board, int n) {
// check upper diagonal
int duprow = row;
int dupcol = col;
while(row >= 0 && col >= 0) {
if(board[row][col] == 'Q') return false;
row--;
col--;
}
col = dupcol;
row = duprow;
while(col>=0) {
if(board[row][col] == 'Q') return false;
col--;
}
row = duprow;
col = dupcol;
while(row<n && col>=0) {
if(board[row][col] == 'Q') return false;
row++;
col--;
}
return true;
}
public:
void solve(int col, vector<string> &board, vector<vector<string>> &ans, int n) {
if(col == n) {
ans.push_back(board);
return;
}
for(int row = 0;row<n;row++) {
if(isSafe1(row, col, board, n)) {
board[row][col] = 'Q';
solve(col+1, board, ans, n);
board[row][col] = '.';
}
}
}
public:
vector<vector<string>> solveNQueens(int n) {
vector<vector<string>> ans;
vector<string> board(n);
string s(n, '.');
for(int i = 0;i<n;i++) {
board[i] = s;
}
solve(0,board, ans, n);
return ans;
}
};