POJ-1321,棋盘问题(DFS)

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

程序代码: 

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
char map[10][10];//存入地图 
int a[10][10];//表示棋盘 
int book[10];//标记棋子的数组 
int n,k,num;
void dfs(int h,int s)//h表示当前所在的行,s表示当前所摆放棋子的数目 
{
	if(s==k)//如果摆放数等于k 
	{
		num++;//方案数加1
		return ;
	}
	if(h>=n)//如果超出棋盘边界直接结束搜索 
		return ;
	for(int i=0;i<n;i++)
	{
		if(a[h][i]&&book[i]==0)//如果当前列没有摆放过棋子,并且当前位置可以摆放
		{
			book[i]=1;//先标记此位置 
			dfs(h+1,s+1);//搜索下一行 
			book[i]=0;//将标记清除 
		}
	}
	dfs(h+1,s);//如果当前行没有可以摆放的位置或者s已经等于k,但是还没有搜索完整个棋盘,将要继续搜索下一行
	return ;
}
int main()
{
	while(cin>>n>>k)
	{
		if(n==-1&&k==-1)
			break;
		num=0;
		memset(a,0,sizeof(a));
		memset(book,0,sizeof(book));
		for(int i=0;i<n;i++)
		{
			getchar();
			for(int j=0;j<n;j++)
			{
				cin>>map[i][j];
				if(map[i][j]=='#')
					a[i][j]=1;//标记可以摆放棋子的位置 
			}
		}
		dfs(0,0);
		cout<<num<<endl;
	}
	return 0;
}
发布了260 篇原创文章 · 获赞 267 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/weixin_43823808/article/details/103795791