poj 1321 dfs

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

这个感觉跟八皇后问题类似，但是就是多了一种k<n的情况，一开始想的是用一个循环遍历每行进行dfs，后来发现不行。参考了下别人的做法，发现就是在普通的基础上加上一个当前行不摆棋子的情况，然后加上一个参数表示摆了几个。。当摆了k个的时候就满足条件了，可以找下一个解。

代码：

#include<iostream>
#include<cstring>
using namespace std;
char map[10][10];
int visit[10];//记录列数 
int n,k;
int m,s;
int count;
void dfs(int u,int num)  //表示当前行  num表示摆了几个 
{
	if(num == k)
	{
		count++;
		return;
	}
	if(u>n) return ;   //避免越界 
	for(int i = 0;i < n;i++)  //当前列 
	{
		if(map[u][i] != '#') continue;
		if(!visit[i])
		{
			visit[i] = 1;
			dfs(u+1,num+1);
			visit[i] = 0;
		}
	}
	dfs(u+1,num); //当前行不摆棋子 
}
int main ()
{
	while(cin >> n >> k)
	{
		if(n==-1 && k==-1) break;
		count = 0;
		memset(visit,0,sizeof(visit));
		for(int i = 0;i < n;i++)
		{
			for(int j = 0;j < n;j++)
			cin >> map[i][j];
		}
		dfs(0,0);
		cout << count << endl;
	}
	return 0;	
}