POJ 1321 棋盘问题(C)回溯

5 篇文章 0 订阅
3 篇文章 0 订阅

Emmm,我又来 POJ 了,这题感觉比上次做的简单点。类似皇后问题。但是稍微做了一点变形,比如棋子数量是不定的。棋盘形状不在是方形等等。

题目链接:POJ 1321 棋盘问题


解题思路

基本思路:从上往下放旗子,每种情况完成后,复盘继续下一种情况。

这里讲一下,void backTrack(int left, int x) 函数。

left 表示还剩的棋子数量,显然如果 left 为 0,说明所有棋子已放完,那么方案数 solution 加 1。

如果不为 0。那么继续检查当前位置的列是否有棋子,如果无棋子,那么当前位置可以放旗子。然后继续递归,棋子数量减 1,行数加 1。如果有棋子,那么悔棋 1 步。继续下一个位置。

C代码
/**
 *	@author wowpH
 *	@date 2019-9-14 19:54:16
 */
#include<stdio.h>
#include<string.h>

#define TRUE 1
#define FALSE 0

#define MAX_N 8		// 矩阵最大为8

#define BOARD TRUE		// 棋盘
#define BLANK FALSE		// 空白

int matrix[MAX_N][MAX_N];// 矩阵,BOARD表示棋盘,BLANK表示空白

int n, k, solution;// solution最终结果

int column[MAX_N];// 每列是否有棋子,TRUE表示有棋子,FALSE表示无棋子

void backTrack(int left, int x) {// 回溯,left表示剩余棋子,x表示当前行
	if (left == 0) {// 无多余棋子
		++solution;	// 方案数加1
		return;
	}

	// 遍历x行及下方的棋盘
	for (int i = x; i < n; ++i) {
		for (int j = 0; j < n; ++j) {
			if (matrix[i][j] == BLANK) {// 空白
				continue;				// 不能放旗子
			}
			if (column[j] == TRUE) {// 第j列有棋子
				continue;			// 不能放旗子
			}

			column[j] = TRUE;			// 当前位置可以放子,设为TRUE
			backTrack(left - 1, i + 1);	// 回溯,棋子数减1,行数加1
			column[j] = FALSE;			// 复盘,设为无子
		}
	}
}

int main() {
	while (scanf("%d %d", &n, &k) && n != -1 && k != -1) {
		getchar();// '\n'

		// 输入棋盘
		for (int i = 0; i < n; ++i) {
			for (int j = 0; j < n; ++j) {
				char ch = getchar();
				if (ch == '.') {
					matrix[i][j] = BLANK;// 空白
				} else if (ch == '#') {
					matrix[i][j] = BOARD;// 棋盘
				}
			}
			getchar();// '\n'
		}

		// 初始化
		memset(column, FALSE, sizeof(column));
		solution = 0;

		backTrack(k, 0);// 回溯

		printf("%d\n", solution);
	}
	return 0;
}
提交结果

在这里插入图片描述

POJ 1321 排兵布阵问题可以使用 DFS 算法求解。 题目要求在一个 n x n 的棋盘上,放置 k 个棋子,其中每行、每列都最多只能有一个棋子。我们可以使用 DFS 枚举每个棋子的位置,对于每个棋子,尝试将其放置在每一行中未被占用的位置上,直到放置了 k 个棋子。在 DFS 的过程中,需要记录每行和每列是否已经有棋子,以便在尝试放置下一个棋子时进行判断。 以下是基本的 DFS 模板代码: ```python def dfs(row, cnt): global ans if cnt == k: ans += 1 return for i in range(row, n): for j in range(n): if row_used[i] or col_used[j] or board[i][j] == '.': continue row_used[i] = col_used[j] = True dfs(i + 1, cnt + 1) row_used[i] = col_used[j] = False n, k = map(int, input().split()) board = [input() for _ in range(n)] row_used = [False] * n col_used = [False] * n ans = 0 dfs(0, 0) print(ans) ``` 其中,row 代表当前尝试放置棋子的行数,cnt 代表已经放置的棋子数量。row_used 和 col_used 分别表示每行和每列是否已经有棋子,board 则表示棋盘的状态。在尝试放置棋子时,需要排除掉无法放置的位置,即已经有棋子的行和列,以及棋盘上标记为 '.' 的位置。当放置了 k 个棋子时,即可计数一次方案数。注意,在回溯时需要将之前标记为已使用的行和列重新标记为未使用。 需要注意的是,在 Python 中,递归深度的默认限制为 1000,可能无法通过本题。可以通过以下代码来解除限制: ```python import sys sys.setrecursionlimit(100000) ``` 完整代码如下:
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值