代码
#include <stdio.h>
#include <malloc.h>
#include <math.h>
#include <stdbool.h>
#include <stdlib.h>
bool place(int* paraSolution, int paraT) {
int j;
for (j = 1; j < paraT; j++) {
if ((abs(paraT - j) == abs(paraSolution[j] - paraSolution[paraT])) || (paraSolution[j] == paraSolution[paraT])) {
return false;
}
}
return true;
}
void backtracking(int* paraSolution, int paraN, int paraT) {
int i;
if (paraT > paraN) {
for (i = 1; i <= paraN; i++) {
printf("%d ", paraSolution[i]);
}
printf("\r\n");
}
else {
for (i = 1; i <= paraN; i++) {
paraSolution[paraT] = i;
if (place(paraSolution, paraT))
backtracking(paraSolution, paraN, paraT + 1);
}
}
}
void nQueen(int paraN) {
int i;
int* solution = (int*)malloc((paraN + 1) * sizeof(int));
for (i = 0; i <= paraN; i++)
solution[i] = 0;
backtracking(solution, paraN, 1);
}
int main() {
nQueen(5);
return 1;
}
运行结果
1 3 5 2 4
1 4 2 5 3
2 4 1 3 5
2 5 3 1 4
3 1 4 2 5
3 5 2 4 1
4 1 3 5 2
4 2 5 3 1
5 2 4 1 3
5 3 1 4 2
代码说明
这段代码是用来解决N皇后问题的。N皇后问题是一个经典的回溯算法问题,即在N×N的棋盘上放置N个皇后,使得它们不能互相攻击,即任何两个皇后都不能处于同一行、同一列或同一斜线上。
1.place函数中整数数组paraSolution代表当前棋盘上的皇后位置,paraT代表当前要放置皇后的位置。它的功能是检查在第paraT行放置一个皇后是否安全
2.backtracking函数中使用回溯算法来尝试放置皇后,如果 paraT 大于 paraN,说明已经放置了所有的皇后
3.nQueen函数初始化棋盘,并调用backtracking函数来开始放置皇后