当前位置: 代码迷 >> 综合 >> HDU4778Gems Fight!(状态压缩DP)
  详细解决方案

HDU4778Gems Fight!(状态压缩DP)

热度:24   发布时间:2023-12-06 19:23:51.0

题目大意:

有B个盒子里面放有G种颜色的宝石,两个人轮流选一个盒子将其中的宝石取出来放到一个锅里,然后其中没有S个相同颜色的宝石,它们就会聚合在一起变成一个魔法石(可能产生多个魔法石且锅里有可能有剩余的宝石),然后本轮的得分就是产生的魔法石的数量,且如果本轮某个人拿到了魔法石的话,那么下一轮他还可以继续选择盒子放入宝石,知道他在某一轮没有拿到魔法石,现在问两个人都采用最优策略的情况下,到最后先拿的那个人的得分与后拿的人的得分的差是多少。


思路:

通过题意我们可以发现,所获得的的魔法石是一定的,也就是说不管按照什么顺序选择,到最后两人拿到的魔法石之和是一定的(这很容易算出来),然后可以发现B最多只有21个,那么我们可以通过状压来表示在某一个状态下先手最多可以拿多少个魔法石。我们可以通过记忆化搜索来解决这个问题,首先在某个特定的状态下,锅里所剩余的宝石是固定的,且还能拿到的魔法石的数量也是一定的。那么我们设dp[flag]表示状态是flag下先手最多可拿的魔法石数量,接下来就是状态转移了。

假设我们选择了第i个盒子((flag>>i)&1==1),则我们可以算可拿到多少个魔法石,设为x个,如果x>0,表示下一轮还是自己选,

则dp[flag]=max(dp[flag],dp[flag^(1<<i)]+x),

否则下一轮是对方选择,

则 dp[flag]=max(dp[flag],left-dp[flag^(1<<i)]),这里的left表示当前状态下还可以拿多少个魔法石。

最后我们求dp[(1<<b)-1]即可。最后还要注意我们要求的答案是两者得分之差,注意转换一下即可。


#include<bits/stdc++.h>
using namespace std;
int dp[1<<21], box[21][8], color[8];
int g, b, s;
int dfs(int stu, int left, int col[])
{if(left==0||stu==0) return 0;if(dp[stu]!=-1) return dp[stu];int ans=0, stu_color[8], status, tmp;memset(stu_color, 0, sizeof(stu_color));for(int i=b-1; i>=0; i--){tmp=0;if((stu>>i)&1){status=stu^(1<<i);int sum=0;for(int j=0; j<g; j++){stu_color[j]=(col[j]+box[i][j]);sum+=stu_color[j]/s;stu_color[j]%=s;}if(sum)tmp=sum+dfs(status, left-sum, stu_color);elsetmp=left-dfs(status, left, stu_color);ans=max(ans, tmp);}}return dp[stu]=ans;
}
int main()
{while(scanf("%d%d%d", &g, &b, &s)){if(!g) break;memset(box, 0, sizeof(box));memset(color, 0, sizeof(color));for(int i=0; i<b; i++){int n, x;scanf("%d", &n);while(n--){scanf("%d", &x);box[i][x-1]++;color[x-1]++;}}int sum=0;for(int i=0; i<g; i++) sum+=color[i]/s;memset(dp, -1, sizeof(dp));int col[8];memset(col, 0, sizeof(col));int ans=dfs((1<<b)-1, sum, col);printf("%d\n", 2*ans-sum);}return 0;
}