zoj 3738 Buy the Pets 状压DP

       有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;
}