数据结构14-N后问题

数据结构14-N后问题

代码

#include <stdio.h>
#include <malloc.h>
#include <math.h>
#include <stdbool.h>
#include <stdlib.h>

// 检查第 paraT 行的皇后是否可以放置在 paraSolution[paraT] 列
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;
}

// 递归回溯法求解 N 皇后问题
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);
        }
    }
}

// 初始化并调用回溯函数求解 N 皇后问题
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);  // 求解 5 皇后问题
	return 1;
}

代码总结

这段代码实现了一个经典的 N 皇后问题的求解算法。N 皇后问题是一个经典的回溯问题，目标是在 ( N \times N ) 的棋盘上放置 N 个皇后，使得她们彼此之间不能相互攻击。具体来说，皇后不能在同一行、同一列或同一对角线上。

1. 头文件包含：

   • #include <stdio.h>：标准输入输出函数库。
   • #include <malloc.h>：动态内存分配函数库。
   • #include <math.h>：数学函数库。
   • #include <stdbool.h>：布尔类型支持。
   • #include <stdlib.h>：通用函数库。

2. place函数：

   • 用于检查在第 paraT 行放置皇后时，是否会与之前放置的皇后发生冲突。
   • 通过比较列差和行差是否相等（对角线冲突）以及列是否相同（列冲突）来判断。

3. backtracking函数：

   • 这是一个递归函数，用于尝试在棋盘上放置皇后。
   • 如果 paraT 大于 paraN，表示所有皇后已经成功放置，打印解。
   • 否则，在每一列尝试放置皇后，并通过调用 place 函数检查是否冲突，如果不冲突则继续递归放置下一行的皇后。

4. nQueen函数：

   • 负责初始化解决方案数组并调用 backtracking 函数开始求解。
   • 动态分配一个数组 solution 用于存储皇后的位置。

5. main函数：

   • 调用 nQueen(5) 来求解 5 皇后问题。

运行结果

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 
 
 
...Program finished with exit code 1
Press ENTER to exit console.