Description
N个bull,M个barn(N,M<=20),每个bull都有几个自己喜欢的barn,要求为每个bull分配一个barn,使得每个bull所分到的barn都是自己喜欢的,且每个barn至多只能容纳一个bull。求合法的分配方案的种数。
Sample Input
3 4
2 1 4
2 1 3
2 2 4
Sample Ouput
4
这一题用状压。
可以将barn的匹配状态看作0,1,即可组成2进制数。
设DP[i][j]为1~i号bull,匹配j状态的barn。
可得DP方程:DP[i][x+d[j]]+=DP[i][x];(d[j]为2^j-1)
代码:
#include<cstdio>
#include<cstring>
using namespace std;
int f[2][1<<20],a[22][22],d[22];
int n,m;
int main()
{
d[1]=1;for(int i=2;i<=20;i++)d[i]=d[i-1]<<1;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
{
int k;scanf("%d",&k);
for(int j=1;j<=k;j++)
{
int x;scanf("%d",&x);
a[i][x]=1;
}
}
for(int j=1;j<=m;j++)if(a[1][j]==1)f[1][d[j]]=1;
int maxx=(1<<m)-1;
int st=1;
for(int i=2;i<=n;i++)
{
st=1-st;memset(f[st],0,sizeof(f[st]));
for(int j=1;j<=m;j++)
{
if(a[i][j]==1)
{
for(int x=0;x<=maxx;x++)
{
if(f[1-st][x]>0)
{
if((x&d[j])==0)
{
f[st][x+d[j]]+=f[1-st][x];
}
}
}
}
}
}
int ans=0;
for(int i=0;i<=maxx;i++)ans+=f[st][i];
printf("%d\n",ans);
return 0;
}