-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathN-Queens II
106 lines (93 loc) · 3.08 KB
/
N-Queens II
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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
class Solution {
private static boolean isvalid(char[][] board, int cr, int cc) {
//left
for(int i = cc; i < board[0].length; i++) {
if(board[cr][i] == 'Q') return false;
}
//rigth
for(int i = cc; i >= 0; i--) {
if(board[cr][i] == 'Q') return false;
}
//up
for(int i = cr; i >= 0; i--) {
if(board[i][cc] == 'Q') return false;
}
//down
for(int i = cr; i < board.length; i++) {
if(board[i][cc] == 'Q') return false;
}
//upper diagonal left
for(int i = cr, j = cc; i >= 0 && j < board[0].length; i--, j++) {
if(board[i][j] == 'Q') return false;
}
//upper diagonal right
for(int i = cr, j = cc; i >= 0 && j >= 0; i--, j--) {
if(board[i][j] == 'Q') return false;
}
//lower left diagonal
for(int i = cr, j = cc; i < board.length && j < board[0].length; i++, j++) {
if(board[i][j] == 'Q') return false;
}
//lower rigth diagonal
for(int i = cr, j = cc; i < board.length && j >= 0; i++, j--) {
if(board[i][j] == 'Q') return false;
}
return true;
}
private static boolean check(char[][] board){
for(int i = 0; i < board.length; i++){
boolean temp = false;
for(int j = 0; j < board[0].length; j++) {
if(board[i][j] == 'Q') {
temp = true;
break;
}
}
if(!temp) return false;
}
return true;
}
private static List<String> convert(char[][] board){
List<String> ans = new ArrayList<>();
for (int i = 0; i <board.length; i++){
String temp = "";
for(int j = 0; j < board[0].length; j++){
temp += board[i][j];
}
ans.add(temp);
}
return ans;
}
private static void solve(List<List<String>> ans, char[][] board, int cr, Set<List<String>> hs) {
if(check(board)){
List<String> temp = convert(board);
if(!hs.contains(temp)) {
ans.add(temp);
hs.add(temp);
}
return;
}
if(cr == board.length) return;
for(int i = 0; i < board[0].length; i++) {
if(isvalid(board, cr, i)) {
board[cr][i] = 'Q';
solve(ans, board, cr + 1, hs);
board[cr][i] = '.';
}
}
}
public int solveNQueens(int n) {
List<List<String>> ans = new ArrayList<>();
char[][] board = new char[n][n];
Set<List<String>> hs = new HashSet<>();
// Arrays.fill(board, '.');
for(int i = 0; i < n; i++)
for(int j = 0; j < n; j++)
board[i][j] = '.';
solve(ans, board, 0, hs);
return ans.size();
}
public int totalNQueens(int n) {
return solveNQueens(n);
}
}