首页 > 代码库 > poj 2441 Arrange the Bulls 状压dp入门

poj 2441 Arrange the Bulls 状压dp入门

题意:

将n头牛和m个栏做匹配,求匹配方案数。

分析:

开始暴搜tle了,还是要用状压dp,dp[i][s]表示前i头牛匹配栏的状态为s时可行的方案数。

代码:

//poj 2441
//sep9
#include <iostream>
using namespace std;
const int maxN=21;
int dp[2][1<<maxN];
int a[maxN][maxN];
int main()
{
	int n,m;		
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;++i){
		scanf("%d",&a[i][0]);
		for(int j=1;j<=a[i][0];++j){
			scanf("%d",&a[i][j]);
			--a[i][j];
		}
	}
	memset(dp,0,sizeof(dp));
	for(int i=1;i<=a[1][0];++i)
		dp[1][1<<a[1][i]]=1;
	int x=0,y=1;
	for(int i=2;i<=n;++i){
		x=y;
		y=x^1;
		memset(dp[y],0,sizeof(dp[y]));
		for(int s=0;s<(1<<m);++s){
			if(dp[x][s])
				for(int j=1;j<=a[i][0];++j){
					int k=(1<<a[i][j]); 				
					if((s&k)==0)
						dp[y][s|k]+=dp[x][s];
				}	
		}
	}
	int ans=0;
	for(int i=0;i<(1<<m);++i)
		ans+=dp[y][i];
	printf("%d",ans);
}


poj 2441 Arrange the Bulls 状压dp入门