思路:题意是有k种类型的物品,列举了m种,每种类型只能选一个,利用公式计算DMG的最大值。
需要遍历多种可能,故用dfs算法来优化,注意当某种类型不存在的情况。解析中使用了一个三维数组,方便计算。
#include <cstdio>
typedef long long ll;
int T,t,n,k,cnt[60],nxt[60];
int sto[60][60][4];
ll biggest;
void dfs(int t,int a,int b,int c,int d){
//截止条件
if(t>k){
ll ans = 1LL*a*b*c*d;
if(ans>biggest){
biggest = ans;
}
return;
}
int num = cnt[t];
if(!num){
dfs(nxt[t],a,b,c,d);
return;
}
//遍历每种可能
for(int i=1;i<=num;i++){
dfs(t+1,a+sto[t][i][0],b+sto[t][i][1],c+sto[t][i][2],d+sto[t][i][3]);
}
}
int main(){
scanf("%d",&T);
while(T--){
scanf("%d%d",&n,&k);
for(int i=1;i<=k;i++)cnt[i]=0;//初始化,不会对之后的测试数据产生影响
for(int i=0;i<n;i++){
scanf("%d",&t);
cnt[t]++;
for(int k=0;k<4;k++){
scanf("%d",&sto[t][cnt[t]][k]);//输入
}
}
t=k+1;
for(int i=k;i;i--){
nxt[i]=t;
if(cnt[i])
t=i;
}
biggest = 0;
dfs(1,100,100,100,100);
printf("%lld\n",biggest);
}
return 0;
}