有n个人,每个人要买一只猫和一只狗做宠物,有的人对某些猫过敏,有些狗和某些猫在一起会打架..现在问若每个人都能买到一只狗和一只猫,并且人猫过敏的组合,猫狗打架的组后都没有出现,一共有多少种可行方案?这题中心点是猫,所以算出人-猫之间有多少种可行方案,在根据每种可行方案猫的状态来确定猫狗的方案数,两个数乘积就是最后的答案。由于N,M,K都很小,所以可以用状压来做。dp1[i][k]表示前i个人,选择的猫的状态是k时,有多少种方案,边界dp1[0][0]=1。之后再dp[n][k]中,找出状态k中有n个1的k,处理处此时选择的猫的编号,在猫-狗之间再做一次DP,同样dp2[i][k]表示前i只猫,选择的狗转台为k。然后扫出来dp[n][k]中转台k包涵n个1的情况,将对应情况累加就是猫-狗搭配的方案...复杂度有点乱,看结构好像是1024*10*10*1024,但实际上内三层循环执行的次数会远小于1024(状态i中恰好包涵n个1的情况不多),所以时间复杂度没问题。
#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <queue>
#include <algorithm>
#include <vector>
using namespace std;
typedef long long ll;
const int maxn=12;
int n,m,p,q,k;
ll dp1[maxn][1<<maxn],dp2[maxn][1<<maxn];
bool g[33][33];
int x,y;
int sta;
vector<int>cat;
int count(int sta,bool op)
{
int res=0;
int cnt=1;
while (sta)
{
if (op && sta&1) cat.push_back(cnt);
res+=(sta&1);
sta>>=1;
cnt++;
}
return res;
}
int main()
{
// freopen("in.txt","r",stdin);
while(~scanf("%d%d%d",&n,&m,&p))
{
memset(g,true,sizeof g);
sta=(1<<m);
scanf("%d",&k);
for (int i=1; i<=k; i++)
{
scanf("%d%d",&x,&y);
g[x][y]=g[y][x]=false;
}
memset(dp1,0,sizeof dp1);
memset(dp2,0,sizeof dp2);
dp1[0][0]=1;
for (int i=1; i<=n; i++)
for (int j=n+1; j<=n+m; j++)
if (g[i][j])
{
for (int k=0; k<sta; k++)
if (dp1[i-1][k])
dp1[i][k|(1<<(j-n-1))]+=dp1[i-1][k];
}
ll ans=0;
for (int i=0; i<sta; i++)
{
cat.clear();
if (count(i,1)==n)
{
memset(dp2,0,sizeof dp2);
ll tmp=0;
dp2[0][0]=1;
for (int ii=1; ii<=n; ii++)
for (int j=n+m+1; j<=n+m+p; j++)
if (g[n+cat[ii-1]][j])
{
for (int k=0; k<(1<<p); k++)
if (dp2[ii-1][k])
dp2[ii][k|(1<<(j-n-m-1))]+=dp2[ii-1][k];
}
for (int k=0; k<(1<<p); k++)
if (count(k,0)==n) tmp+=dp2[n][k];
ans+=dp1[n][i]*tmp;
}
}
printf("%lld\n",ans);
}
return 0;
}