码迷,mamicode.com
首页 > 其他好文 > 详细

POJ 2441 Arrange the Bulls

时间:2016-05-08 18:27:29      阅读:209      评论:0      收藏:0      [点我收藏+]

标签:

状压DP。

dp[i][j]表示前i头牛选完,状态j中为1的篮子选完的方案数。

#include <cstdio>
#include <cstring>
#include <vector>
using namespace std;

const int maxn=21;
int n,m;
vector<int> g[maxn];
int dp[2][1<<maxn];

int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        for(int i=0;i<maxn;i++) g[i].clear();
        for(int i=1;i<=n;i++)
        {
            int p; scanf("%d",&p);
            while(p--)
            {
                int tmp; scanf("%d",&tmp); tmp--;
                g[i].push_back(tmp);
            }
        }

        int f=0;
        for(int i=0;i<=(1<<m);i++) dp[f][i]=0;

        for(int i=0;i<g[1].size();i++)
            dp[f][1<<g[1][i]]=1;

        for(int i=2;i<=n;i++)
        {
            f=f^1;
            for(int j=(1<<m)-1;j>=0;j--)
            {
                dp[f][j]=0;
                if(dp[f^1][j]==0) continue;//这个优化不加会TLE
                for(int k=0;k<g[i].size();k++)
                {
                    if(((1<<g[i][k])&j)==0)
                        dp[f][(1<<g[i][k])|j]+=dp[f^1][j];
                }
            }
        }

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

 

POJ 2441 Arrange the Bulls

标签:

原文地址:http://www.cnblogs.com/zufezzt/p/5471012.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!