poj1312棋盘问题(dfs)

本文介绍了一种使用深度优先搜索解决N皇后问题的方法,通过递归遍历棋盘的每一行,确保放置的皇后不在同一行或同一列,以此来计算放置k个皇后时的所有可能方案。

题目:http://poj.org/problem?id=1321

Description:

在n*n的棋盘上摆棋子,棋子没有区别,要求任意两个棋子不能放在同一行或同一列,求摆放k个棋子时的方案数c为多少。

Solution:

#include<iostream>
#include<stdio.h>
using namespace std;
char Map[9][9];
int Book[9];
int n;//棋盘大小 n*n
int k;//棋子数
int num;//已放棋子数
int C;//方案数

//从第一行第一列到第一行第二列......到第一行第n列遍历
//再从第二行第一列到第二行第二列......到第二行第n列遍历
//再从第三行...

void dfs(int line)
{
	if(num == k)
	{
		C++;
		return;
	}

	if(line >= n)
	{
		return;//如果走到了最后一行,那么返回上一行
	}

	for(int column = 0; column < n; column++)
	{
		if(Book[column] == 0 && Map[line][column] == '#')
		{
			Book[column] = 1;
			num++;
			dfs(line+1);
			Book[column] = 0;
			num--;
		}
	}

	//走到这里说明走到了最后一列,但是没有走到最后一行
	dfs(line+1);//从下一行开始继续寻找可行方案
}
int main()
{
	while(cin >> n >> k)
	{
		C = 0;
		if(n == -1 && k == -1)
		break;
		for(int i = 0; i < n; i++)
		{
			for(int j = 0; j < n; j++)
			{
				cin >> Map[i][j];
			}
		}
		dfs(0);
		cout << C << endl;
 	}
        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、付费专栏及课程。

余额充值