HDU4778 Gems Fight!（记忆化搜索+博弈）

传送门:http://acm.hdu.edu.cn/showproblem.php?pid=4778

题目大意：有B个包裹，里面有各种颜色的GEM，共有G个颜色。Alice和Bob轮流挑选包裹放到一个地方，如果挑选出来的同种颜色的GEM超过S个，当前回合者可以得分，每超过S个得一分。如果在回合内得分了，可以额外进行一个回合。问Alice的得分减去Bob的得分最大是多少。

#include<iostream>
#include<algorithm>
#include<string>
#include<map>//ll dx[4]={0,0,-1,1};ll dy[4]={-1,1,0,0};
#include<set>//
#include<vector>
#include<cmath>
#include<stack>
#include<string.h>
#include<stdlib.h>
#include<cstdio>
#define mod 1e9+7
#define ll long long
using namespace std;
#define REP(i,n) for(int i=0;i<(n);++i)
struct node {
int c[10]; //一个包里某颜色个数
}Bag[30];
int b,g,s,now[10],sg[1<<22];
int judge(int k){
int total = 0;
REP(i,g+1){
now[i]+=Bag[k].c[i];
total+=now[i]/s; //每超过s个得一分
now[i]%=s; //该颜色(i)剩下的个数
}
return total; //返回得到的分数
}
int dfs(int dp){
if(sg[dp] != -1)
return sg[dp];
int tep[10],ans = -100000000; //结果有可能是负的
for(int i=1;i<=g;++i)
tep[i] = now[i]; //now[i]表示现在第i种颜色的有几个了
for(int i=0;i<b;++i){ //遍历背包
int k = 1 << i;
if(!(dp & k)){ //dp这个状态里第k个包没取
int total = judge(i); //判断当前这个人能得多少分
if(total > 0) //得分了
ans = max(ans,total + dfs(dp|k)); //下一轮还是自己选 ，dp|k表示算取了第k个包
else
ans = max(ans,total - dfs(dp|k)); //下一轮对方选
for(int j=1;j<=g;++j)
now[j] = tep[j]; //注意，每进行一次上面的dfs，now[]值会变
}
}
return sg[dp] = ans;
}

void init() {
int k = 1 << b;
memset(sg,-1,sizeof(sg));
sg[k-1] = 0; //记忆化搜索最终状态，所有包都取完
memset(now,0,sizeof(now));
memset(Bag,0,sizeof(Bag));
}
int main() {
int n,m,t,cs=0;
while(~scanf("%d%d%d",&g,&b,&s)) { //颜色数、包裹数、回合数
if(!b && !g && !s)
break;
init();
REP(i,b){
scanf("%d",&n); //第i个包裹里有n种颜色
while(n--){
scanf("%d",&m); //颜色编号
Bag[i].c[m]++;
}
}
printf("%d\n",dfs(0)); //0表示一个包都没取的状态
}
return 0;
}
#include<iostream>
#include<algorithm>
#include<string>
#include<map>//int dx[4]={0,0,-1,1};int dy[4]={-1,1,0,0};
#include<set>//
#include<vector>
#include<cmath>
#include<stack>
#include<string.h>
#include<stdlib.h>
#include<cstdio>
#define mod 1e9+7
#define ll long long
using namespace std;
struct node{
int c[10];
}x[30];
int g,b,k,m,a;
int now[30];
int dp[(1<<22)];
int ss;
int dfs(int p){
if(dp[p]!=-1)
return dp[p];
int tmp[10]; //注意是内部函数，需要定义在里面
for(int i=1;i<=g;++i){
tmp[i]=now[i];
}
int ss=-10000000;
for(int i=0;i<b;++i){
if(p&(1<<i))
continue;
int cnt=0;
for(int j=1;j<=g;++j){
now[j]+=x[i].c[j];
cnt+=now[j]/k;
now[j]%=k;
}
if(cnt>0)
ss=max(ss,cnt+dfs(p|(1<<i)));
else
ss=max(ss,cnt-dfs(p|(1<<i)));
for(int j=1;j<=g;++j)
now[j]=tmp[j];
}
dp[p]=ss;
return dp[p];
}
int main(){
while(cin>>g>>b>>k){
if(g+b+k==0)
break;
memset(x,0,sizeof(x));
memset(now,0,sizeof(now));
for(int i=0;i<b;++i){
cin>>m;
for(int j=0;j<m;++j){
cin>>a;
x[i].c[a]++;
}
}
memset(dp,-1,sizeof(dp));
dp[(1<<b)-1]=0;
cout<<dfs(0)<<endl;
}
return 0;
}