POJ1321 棋盘摆放问题（DFS）

每日打卡（1/2）  

传送连接：点击打开链接

题目大意：

    给你一个n*n的棋盘，其中只有标#的格子能放棋子，且两个棋子不能位于棋盘的同一行或同一列，要求摆k个棋子，求方案数。

大致思路：

    裸DFS，和n皇后思路相同。以行数作为基准，如果当前列有符合要求的格子，可以采取放或不放两种方法，最后求和。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cstdlib>
#include<algorithm>
using namespace std;
const int maxn = 10;

int n,k,a[maxn][maxn];
long long ans;
bool vis[maxn];

void dfs(int x,int tot)
{
	if(tot<k&&x>=n) return;
	if(tot==k) {
		ans++;
		return;
	}
	else {
		for(int i=0;i<n;i++)
		{
			if(a[x][i]&&!vis[i]) {
				vis[i] = 1;
//				cout<<i<<endl;
				dfs(x+1,tot+1);
				vis[i] = 0;
			}
		}
		dfs(x+1,tot);
	}
}
int main()
{
	while(scanf("%d%d",&n,&k)!=EOF&&n!=-1&&k!=-1)
	{
		char s;
		ans = 0;
		memset(a,0,sizeof(a));
		memset(vis,0,sizeof(vis));
		for(int i=0;i<n;i++)
		 for(int j=0;j<n;j++)
		 {
		 	cin>>s;
		 	if(s=='#') a[i][j] = 1;
		 	else a[i][j] = 0;
		 }
		dfs(0,0);
		cout<<ans<<endl; 
	}
	return 0;
}

这个解法会TLE，这时候我们可以考虑剪枝。

不难想到，如果总共有5行，当前已经搜到了第4行，却还有3个棋子没有摆，那么这种情况就可以剪枝掉。

因此，我们在dfs的开头加上一行：

if(k-tot>n-x) return;

如果剩下的棋子超过剩下的行数，那么就剪枝掉。

所以就AC了。