POJ 1321 棋盘问题（DFS）

传送门：点击打开链接

棋盘问题

Time Limit: 1000MS		Memory Limit: 10000K
Total Submissions: 47261		Accepted: 22851

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皇后问题，简单的DFS就可以解决。将摆好棋子的数目来DFS，将棋子从0到n-1列放置，若符合题目要求，则进行下一个棋子的放置，知道放置好的棋子数等于棋子总数，此时总数加一，最后输出总数，具体实现看代码。

代码实现：

/* N皇后问题 */ 
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>

using namespace std;

char map[15][15];   //记录棋盘 
int ans,sum,n,k;    //ans表示当前摆好的棋子数，sum表示总数 
bool visit[15];     //标记某列是否放置了棋子 

void dfs(int x)
{
	int i;
	if(ans==k)      //结束dfs条件，所有棋子已经摆好 
	{
		sum++;      //总数加1 
		return ;
	}
	
	if(x>=n)        //越界判断 
	return ;
	
	for(i=0;i<n;i++)    //每列都放一次 
	{
		if(map[x][i]=='#'&&!visit[i])   //可以放置 
		{
			visit[i]=true;  //标记 
			ans++;          //棋子数++ 
			dfs(x+1);       //递归下一个棋子 
			ans--;          //状态还原 
			visit[i]=false;
		}
	}
	dfs(x+1);   //递归下一个 
	
}

int main()
{
	int i;
	while(cin>>n>>k)
	{
		getchar();
		if(n==-1&&k==-1)
		break;
		
		memset(visit,false,sizeof(visit));
		memset(map,0,sizeof(map));
		
		sum=0;ans=0;
		for(i=0;i<n;i++)
		cin>>map[i];
		
		dfs(0);
		cout<<sum<<endl;
	}
	return 0;
}