2013杭州站I - Gems Fight!（状态dp）

题目地址

题目大意：现有g（最多为8）种颜色（1、2......g）的宝石，有b（最多21）个包（1、2......b），每个包里放有一定数目的宝石，2个人轮流选一个包，将包里的宝石倒入锅里，若锅里有s个相同颜色的宝石，则可以得到一块魔石，若该选手得到了魔石，他可以继续再选择一个包里的宝石放入锅里，继续得到魔石，直到不能得到魔石则该另一个人选包，问先手得到的魔石数目与后手得到的魔石数目之差

解题思路：将选择包的情况看成一种状态，0表示已经选过了，1表示可以选，dp[i]表示先手在i状态的情况下最多能获得的魔石的数目

#include <bits/stdc++.h>

using namespace std;

const int N = 1 << 21;

bool vis[N];

int n,g,s,dp[N],score[N],a[22][9];

void get_score()
{
    int sum[9];///记录每种颜色的个数
    for(int mask=0;mask<(1<<n);mask++) ///对于每种取包状态，0表示已经取过，1表示还未取过（还可以取）
    {
        memset(sum,0,sizeof(sum));
        for(int i=0;i<n;i++) ///对于每个位置的包
            if(!(mask >> i & 1)) ///如果已经取过
                for(int j=1;j<=g;j++) ///求每种颜色和（对可以取的所有包）
                    sum[j] += a[i][j];
        int ans = 0;
        for(int j=1;j<=g;j++)
            ans += sum[j] / s;
        score[mask] = ans; ///表示mask状态的所有取过的包所得到的宝石数
    }
}

int dfs(int mask)
{
    if(vis[mask]) ///对于mask状态已求过
        return dp[mask];
    vis[mask] = true; ///标记求他
    int ret = 0;
    for(int i=0;i<n;i++)///对于n个包
        if(mask >> i & 1) ///状态mask的该位还未取过
        {
            int __mask = mask ^ (1 << i); ///仅把i位变为0，表示选择了该位，不能再选择其了
            int _delta = score[__mask] - score[mask];///选过-没选过该位包的宝石数量
            if(_delta > 0) ///有宝石，继续该他
                ret = max(ret,_delta + dfs(__mask));
            else///没有宝石，换人
                ret = max(ret,score[0] - score[mask] - dfs(__mask));///取完的宝石数
        }
    return dp[mask] = ret;///返回先手在mask状态下的最优宝石数
}

int main()
{
    while(~scanf("%d%d%d",&g,&n,&s),g||n||s)
    {
        memset(vis,false,sizeof(vis));///初始化每种状态是否求过
        memset(a,0,sizeof(a));///每个包的每种颜色的数量
        for(int i=0;i<n;i++)
        {
            int num,color;
            scanf("%d",&num);
            while(num--)
            {
                scanf("%d",&color);
                a[i][color]++;
            }
        }
        get_score();///求得所有状态可以取的最大宝石数
        printf("%d\n",2*dfs((1<<n)-1)-score[0]);
    }
    return 0;
}