Poj-2441 Arrange the Bulls //状压dp

题目链接

http://poj.org/problem?id=2441

题意

有 $n$ 头牛，$m$ 个谷仓，每头牛可以去指定的几个谷仓打篮球，在一个谷仓只能容纳一头牛的前提下，问有多少种方案使得这 $n$ 头牛都有地方打篮球

解析

首先肯定想到暴力搜索的做法，就是先给一头牛分配一个谷仓，然后在与前 $i-1$ 头牛谷仓不冲突的情况下，给第 $i$ 头牛分配谷仓，直到分配到最后一头牛，合法则对答案贡献加 $1$。但是这样做的最坏时间复杂度为：$O(m^n)$，稳稳的超时。
所以只能用状压dp来做了，如果我们定义集合 $S$ 为：使用过的谷仓集合。也就是二进制表示位的第 $i$ 位为 $1$ ，则说明第 $i$ 个谷仓使用过了。
我们不妨从第一头牛开始分配，定义 $dp[i][S]$：前 $i$ 头牛分配的谷仓集合为 $S$ 时的方案数。
则不难得出递推式：$dp[i][S^{\prime }]={\sum }_{S|k=S^{\prime }}dp[i-1][S]$ (其中 $k$ 为第 $i$ 头牛可以去的且和前 $i-1$ 头牛去的都不同的谷仓编号)
则最终 $\sum dp[n][S]$ 即为最终结果，时间复杂度：$O(n\ast m\ast 2^m)$

坑点

如果不优化空间会MLE，所以可以用01滚动或者就地滚动优化一下空间，注意就地滚动时，要从大集合向小集合递推，也就是for循环方向要从大到小枚举，可以联想背包的优化理解下为什么

代码

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
typedef long long ll;
typedef pair<int,ll> P;
const int maxn=25;
const int inf=0x3f3f3f3f;
int m,n;
int g[maxn];
int dp[1<<20];
int main(){
    cin>>n>>m;
    for(int i=1,x;i<=n;i++){
        cin>>x;
        for(int j=1,y;j<=x;j++){//把每头牛可以去的谷仓也压缩成一个二进制数
            scanf("%d",&y);y--;
            g[i]|=(1<<y);//比如:1 4 -> 1001
        }
    }
    for(int i=0;i<m;i++){//先算第一头牛的
        if((1<<i)&g[1])dp[1<<i]=1;
    }
    int len=1<<m;
    for(int i=2;i<=n;i++){
        for(int j=len-1;j>=1;j--){//大集合->小集合,方向不可变
            if(!dp[j])continue;//集合不存在就跳过
            for(int k=0;k<m;k++){//看当前牛可以去的谷仓
                if(!(g[i]&(1<<k)))continue;//当前牛不可以去
                if((1<<k)&j)continue;//前i-1头牛用过此谷仓
                dp[j|(1<<k)]+=dp[j];
            }
            dp[j]=0;//用过要及时清零
        }
    }
    int ans=0;
    for(int i=1;i<len;i++){
        ans+=dp[i];
    }
    printf("%d\n",ans);
    return 0;
}