poj 1321 棋盘问题 DFS

棋盘问题

Time Limit: 1000MS Memory Limit: 10000K
Total Submissions: 36999 Accepted: 18227

Description

在一个给定形状的棋盘(形状可能是不规则的)上面摆放棋子,棋子没有区别。要求摆放时任意的两个棋子不能放在棋盘中的同一行或者同一列,请编程求解对于给定形状和大小的棋盘,摆放k个棋子的所有可行的摆放方案C。

Input

输入含有多组测试数据。
每组数据的第一行是两个正整数,n k,用一个空格隔开,表示了将在一个n*n的矩阵内描述棋盘,以及摆放棋子的数目。 n <= 8 , k <= n
当为-1 -1时表示输入结束。
随后的n行描述了棋盘的形状:每行有n个字符,其中 # 表示棋盘区域, . 表示空白区域(数据保证不出现多余的空白行或者空白列)。

Output

对于每一组数据,给出一行输出,输出摆放的方案数目C (数据保证C<2^31)。

Sample Input

2 1
#.
.#
4 4
...#
..#.
.#..
#...
-1 -1

Sample Output

2
1

这个题还是比较有意思的,首先,它给出的棋盘不是规整的,其次,给的数据也有两种,一种是k=n,另一种是k<n;
第一种还是很好做的,但是第二种就有些麻烦了。
但由于数据比较小,所以还是很好做的。
关于DFS的思路就不说了,以为也比较抽象,所以直接上代码。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int n, k;
char a[10][10];
int col[10];
int ans = 0;
void dfs(int st, int num)
{
	for(int i = 1;i <= n;i ++)
	{
		if(a[st][i] == '#' && col[i] == 0)
		{
			if(num == 1)
				ans++;
			else
			{
				col[i] = 1;
				for(int j = st + 1;j <= n - num + 2;j ++)
					dfs(j, num - 1);
				col[i] = 0;
			}
		}
	}
}
int main()
{
	while(scanf("%d%d", &n, &k) != EOF)
	{
		ans = 0;
		if(n == -1 && k == -1)
			return 0;
		for(int i = 1;i <= n;i ++)
			for(int j = 1;j <= n;j ++)
				cin>>a[i][j];
		for(int i = 1;i <= n;i ++)
			col[i] = 0;
		for(int i = 1;i <= n - k + 1;i ++)
		{
			dfs(i, k);
		}
		printf("%d\n", ans);
	}
	return 0;
}

 
 

上一篇
下一篇