小 A 的卡牌游戏
小 A 最近沉迷于一款名为 Hearthverse 的卡牌游戏。在这款游戏中,卡被分为了三个种类(随从、法术和魔法阵),在组卡时,这款游戏严格规定了卡组中每种卡牌的数量,具体来说,一副 n 张卡的卡组需要包含恰好 a 张随从卡, b 张法术卡和 c 张魔法阵卡,并且 a+b+c=n。
在游戏中,有一个叫做 “3pick” 的组卡竞技方式,游戏会给出 n
次三选一的机会,三张卡分别来自三个种类。每次玩家需要从三张卡中选出一张卡加入自己的卡组。为了使卡组强度尽量高,小 A 给每张卡设定了一个强度值,而卡组的强度就是所有卡强度值的和,他希望他最后组出的卡组的强度可以尽量高。
solution
设三元组为(x,y,z)
如果只有二维,直接按照x-y排序即可,现在有三维,只需再套上一个dp即可。
#include<cstdio>
#include<algorithm>
#define fo(i,a,b) for (int (i)=(a);(i)<=(b);(i)++)
using namespace std;
const int N=5e3+5;
typedef long long ll;
struct node{
ll x,y,z;
};
ll f[N],g[N];
int a,b,c,n;
node t[N];
bool cmp(node a,node b){
return a.x-a.y > b.x-b.y;
}
int main(){
// freopen("data.in","r",stdin);
scanf("%d %d %d %d",&n,&a,&b,&c);
fo(i,1,n) {
scanf("%lld %lld %lld",&t[i].x, &t[i].y, &t[i].z);
}
sort(t+1,t+n+1,cmp);
fo(i,1,c) f[i]=-1e9;
f[0]=0;
fo(i,1,n) {
fo(j,0,c) g[j]=-1e9;
fo(j,max(0,i-a-b),min(i,c)) {
if (j) {
g[j]=max(g[j],f[j-1]+t[i].z);
}
if (j+a>=i) g[j]=max(g[j],f[j]+t[i].x);
else g[j]=max(g[j],f[j]+t[i].y);
}
fo(j,0,c) f[j]=g[j];
}
printf("%lld",f[c]);
return 0;
}