-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathChessboard-Queens.java
89 lines (79 loc) · 2.46 KB
/
Chessboard-Queens.java
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
//n*n chessboard has n queens in it which are to be placed in such a way that each of the queen is safe by using the logic of backtraking
import java.util.*;
class Queens{
public boolean isSafe(int row, int col, char[][] board) {
for(int j=0; j<board.length; j++) {
if(board[row][j] == 'Q') {
return false;
}
}
//vertical
for(int i=0; i<board.length; i++) {
if(board[i][col] == 'Q') {
return false;
}
}
//upper left
int r = row;
for(int c=col; c>=0 && r>=0; c--, r--) {
if(board[r][c] == 'Q') {
return false;
}
}
//upper right
r = row;
for(int c=col; c<board.length && r>=0; r--, c++) {
if(board[r][c] == 'Q') {
return false;
}
}
//lower left
r = row;
for(int c=col; c>=0 && r<board.length; r++, c--) {
if(board[r][c] == 'Q') {
return false;
}
}
//lower right
for(int c=col; c<board.length && r<board.length; c++, r++) {
if(board[r][c] == 'Q') {
return false;
}
}
return true;
}
public void saveBoard(char[][] board, List<List<String>> allBoards) {
String row = "";
List<String> newBoard = new ArrayList<>();
for(int i=0; i<board.length; i++) {
row = "";
for(int j=0; j<board[0].length; j++) {
if(board[i][j] == 'Q')
row += 'Q';
else
row += '.';
}
newBoard.add(row);
}
allBoards.add(newBoard);
}
public void helper(char[][] board, List<List<String>> allBoards, int col) {
if(col == board.length) {
saveBoard(board, allBoards);
return;
}
for(int row=0; row<board.length; row++) {
if(isSafe(row, col, board)) {
board[row][col] = 'Q';
helper(board, allBoards, col+1);
board[row][col] = '.';
}
}
}
public List<List<String>> solveNQueens(int n) {
List<List<String>> allBoards = new ArrayList<>();
char[][] board = new char[n][n];
helper(board, allBoards, 0);
return allBoards;
}
}