状态压缩DP+poj2441

题意：有n头牛，m个房间，每头牛有不同的喜欢的房间，把他们安排好，一共有多少种方法

思路：dp[i]表示当前状态一共有多少种方法往下递推。

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
int N,M;
int dp[(1<<20)+5],map[25][25];
int main()
{
    #ifndef ONLINE_JUDGE
        freopen("in.txt","r",stdin);
    #endif
    int p,barn;
    while(cin>>N>>M)
    {
        memset(map,0,sizeof(map));
        for(int i=0;i<N;i++)
        {
            cin>>p;
            for(int j=0;j<p;j++)
            {
                cin>>barn;
                barn--;
                map[i+1][barn]=1;
            }
        }
        memset(dp,0,sizeof(dp));
        dp[0]=1;
        for(int i=0;i<N;i++)
        {
            for(int j=(1<<M)-1;j>=0;j--)
            {
                if(dp[j]==0) continue;
                for(int k=0;k<M;k++)
                {
                    if(j&(1<<k)) continue;
                    if(map[i+1][k]==0) continue;
                    dp[j|(1<<k)]+=dp[j];
                }
                dp[j]=0;//这里中间状态清零，因为下面要求和。
            }
        }
        int ans=0;
        for(int i=0;i<(1<<M);i++)
        ans+=dp[i];
        cout<<ans<<endl;
    }
    return 0;
}