poj2441 Arrange the Bulls (状态压缩dp)

问题描述

有N只牛和M个谷仓(N和M小于等于20)
每只牛只喜欢在特定的谷仓里面
给出每只牛喜欢的谷仓
问所有牛都在自己喜欢的谷仓里的方案数

输入

在输入的第一行中包含两个整数N和M (1 <= N <= 20, 1 <= M <= 20)。然后是N行。第i行首先包含一个整数P (1 <= P <= M)，它表示我喜欢在其中玩耍的谷仓奶牛的数量。然后是P个整数，它给出了P个谷仓的数量。

输出

在一行中打印一个整数，这是解决方案的数量。（保证答案在int范围内）

分析：

1表示谷仓有牛了，0表示谷仓没有牛

用d[ j ]表示j状态下的方案数，状态 j 的二进制中1的个数就是该状态下牛的数量
放牛的顺序是不影响结果的，不妨从第一头牛开始。

轮到插入第i头牛的时候
遍历所有状态，找到其中只有(i-1)头牛的状态（即找到二进制数中1的个数为(i-1)的状态）
判断这个状态下能否插入当前牛(遍历当前牛喜欢的每一个谷仓)，如果可以，则
d[插入之后的状态]+=d[插入之前的状态]

另外0头牛的时候方案数为1，即d[0]=1;

code:

#include<iostream>
#include<cstdio>
#include<vector>
using namespace std;
const int maxm=23;
int n,m;
vector<int>like[maxm];//存放每只牛的可行位置
int d[1<<maxm];
int one(int x){//统计二进制下1的个数
    int ans=0;while(x){x&=(x-1);ans++;}return ans;
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        int k;
        scanf("%d",&k);
        while(k--){
            int t;
            scanf("%d",&t);
            like[i].push_back(1<<(t-1));//可行位置
        }
    }
    d[0]=1;
    for(int i=1;i<=n;i++){//放第i头牛
        for(int j=0;j<(1<<m);j++){//所有状态
            if(one(j)!=(i-1))continue;//如果1的个数不是i-1个就跳过
            for(int k=0;k<(int)like[i].size();k++){//遍历当前牛喜欢的所有谷仓的位置
                if((like[i][k]&j)==0){//如果能插入
                    d[like[i][k]|j]+=d[j];//之后的状态用或运算|  求出
                }
            }
        }
    }
    int ans=0;
    for(int j=1;j<(1<<m);j++){//统计答案
        if(one(j)==n){
            ans+=d[j];
        }
    }
    printf("%d\n",ans);
    return 0;
}