POJ1321 棋盘问题(DFS)

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_42391248/article/details/82049505

棋盘问题

Time Limit: 1000MS   Memory Limit: 10000K
Total Submissions: 66596   Accepted: 31765

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

N皇后类型的题,模仿着N皇后DFS很好写。唯一需要注意的就是读题呀。。。#代表棋子能下的地方。

#include<iostream>
#include<string>
#include<cstring>
using namespace std;
char map[10][10];
int k,n,ans;
int vis[100];//vis[i]:第i行vis[i]列有棋子 
int jc(int hang)
{
	for(int i=0;i<hang;i++)
		if(vis[hang]==vis[i])
			return 0;		
	return 1;
}
void DFS(int hang,int num)
{
	if(num==k)
	{
		ans++;
		return;
	}
	if(hang==n)
		return;			
	int flag=0;
	for(int i=0;i<n;i++)
		if(map[hang][i]=='.')
			flag++;
	if(flag==n)
		DFS(hang+1,num);
	else
	{
		for(int i=0;i<n;i++)
		{
			if(map[hang][i]=='.')
				continue;
			vis[hang]=i;
			if(jc(hang))
				DFS(hang+1,num+1);
			vis[hang]=-1;			
		}
		DFS(hang+1,num);
	}			
}
int main()
{
	while(cin>>n>>k&&n!=-1&&k!=-1)
	{
		for(int i=0;i<n;i++)
			for(int j=0;j<n;j++)
				cin>>map[i][j];
		memset(vis,-1,sizeof(vis));
		ans=0;
		DFS(0,0);
		cout<<ans<<endl;	
	}
	return 0;	
} 

猜你喜欢

转载自blog.csdn.net/qq_42391248/article/details/82049505