-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathseq_n-queens.c
101 lines (78 loc) · 1.87 KB
/
seq_n-queens.c
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
#include "seq_n-queens.h"
int numSol = 0;
void printSolution(int board[][N])
{
for (int i = 0; i < N; i++)
{
for (int j = 0; j < N; j++)
printf(" %d ", board[i][j]);
printf("\n");
}
printf("\n");
}
bool isSafe(int board[N][N], int row, int col)
{
int i, j;
/* Check this row on left side */
for (i = 0; i < col; i++)
if (board[row][i])
return false;
/* Check upper diagonal on left side */
for (i=row, j=col; i>=0 && j>=0; i--, j--)
if (board[i][j])
return false;
/* Check lower diagonal on left side */
for (i=row, j=col; j>=0 && i<N; i++, j--)
if (board[i][j])
return false;
return true;
}
bool recursiveQueenPlacement(int board[N][N], int col)
{
// Count each solution twice exept for middle row if odd board
if (col >= N) {
if(N % 2 == 1){
if(board[N/2][0] == 0 ){
numSol++;
}
} else {
numSol++;
}
numSol++;
return true;
}
//Place queen for each row in column
for (int i = 0; i < N; i++)
{
// if legit placement
if ( isSafe(board, i, col) )
{
//place queen
board[i][col] = 1;
// recur to place queen next column
recursiveQueenPlacement(board, col + 1);
board[i][col] = 0;
}
}
return false;
}
int main(int argc, char* argv[]) {
n = strtol(argv[1], NULL, 10);
int problems = N/2 + N%2;
int boards[problems][N][N];
for(int i = 0; i < problems; i++){
for(int j = 0; j < N; j++){
for(int k = 0; k < N; k++){
boards[i][j][k] = 0;
}
}
}
for(int i = 0; i < problems; i++){
boards[i][i][0] = 1;
}
for(int i = 0; i < problems; i++) {
recursiveQueenPlacement(boards[i], 1);
}
printf("%d\n", numSol);
return 0;
}