poj 2441 Arrange the Bulls (状态压缩dp+滚动数组)

题意：

n头牛和m个barn，每头牛有自己喜欢的p个barn(1<=p<=m)，问有多少种安排n头牛的方法。

 

分析：

dp[i][j]表示i头牛在j状态下的方法数。

dp[i][j] = sigma(dp[i-1][k])  (k从0到 1<<m)。

要用滚动数组优化，否则dp[20][1<<20]会MLE！

 

注意滚动数组的清空= =！

#include<stdio.h>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>

using namespace std;

typedef long long ll;

vector<int> p[25];
int dp[2][1<<20],n;

int main()
{
    int m,x,y;
    while(~scanf("%d%d",&n,&m))
    {
        for(int i=1;i<=n;i++)
        {
            p[i].clear();
            scanf("%d",&x);
            while(x--)
            {
                scanf("%d",&y);
                p[i].push_back(y);
            }
        }
        memset(dp,0,sizeof(dp));

        int all = (1<<m)-1;
        dp[0][0] = 1;
        int t;

        for(int i=1;i<=n;i++)
        {
            for(int k=0;k<=all;k++)  //每种状态访问一次
            {
                if(!dp[(i-1)&1][k]) continue; //只有前i-1头牛的k状态有值的情况下才能确定第i头牛的放法
                for(int j=0;j<p[i].size();j++)
                {
                    t = p[i][j]-1;
                    if((k>>t)&1) continue; //如果k状态下这个位置已经有牛则不能放
                    dp[i&1][k|(1<<t)] += dp[(i-1)&1][k];
                }
            }
            memset(dp[(i-1)&1],0,sizeof(dp[(i-1)&1]));  //清空滚动数组下次要用的一维
        }

        int ans = 0;
        for(int i=1;i<=all;i++)
        {
            ans+=dp[n&1][i];
        }
        printf("%d\n",ans);
    }
    return 0;
}