(牛客14602)xinjun与阴阳师(01背包变型)
链接:https://ac.nowcoder.com/acm/problem/14602
来源:牛客网
题目描述:
xinjun是各类手游的狂热粉丝,因随手一氪、一氪上千而威震工大,现在他迷上了阴阳师。xinjun玩手游有一个习惯,就是经过层层计算制定出一套方案来使操作利益最大化(因此即使有扫荡券也不用,故称圣雄肝帝)。已知阴阳师有N个模式可以操作,模式i有ai种操作,但每种模式每日只能选用一种操作,可以不选。操作j能收益vj,但需要花费体力wj点。xinjun每日拥有体力M点,求他每日最多能得到多少收益。
输入描述:
第一行一个正整数T(T<=10),表示共有T组数据。
对于每组数据,第一行两个正整数N,M(1<=N,M<=1000)。
接下来N段数据,每段第一行一个正整数ai(1<=ai<=1000),第二行ai个正整数vj(1<=vj<=1000),第三行ai个正整数wj(1<=wj<=1000)。
每组数据ai之和不大于104。
输出描述:
对每组数据输出一行,即xinjun每日最多能得到多少收益。
输入
1
3 10
2
2 3
3 2
2
1 1
3 4
1
5
5
输出
9
这道题本质上还是01背包问题,只不过在此基础上又增加了一重循环(a[i]),多读几遍题目,把输入的数据先给屡清楚,只要输入的数据顺序懂了,那么按照01背包的思路去解就可以了,在此我推一下之前写的01背包模板题思路:01背包问题(动态规划)+一维数组的优化
下面是AC代码:
#include<bits/stdc++.h>
using namespace std;
int a[1010],v[1010][1010],w[1010][1010];
int dp[1010];
int main(){
int t,n,m,i,j,k;
cin>>t;//t组数据
while(t--){
memset(dp,0,sizeof(dp));//每次输入前先重置dp[],v[]和w[]没有必要重置
cin>>n>>m;//n个模式,m点体力
for(i=1;i<=n;i++){
cin>>a[i];//每个模式下有a[i]种操作
for(j=1;j<=a[i];j++){
cin>>v[i][j];//收益
}
for(j=1;j<=a[i];j++){
cin>>w[i][j];//消耗的体力
}
}//-------------------以上是输入的数据-----------------
//三重循环,思路还是01背包的思路
for(i=1;i<=n;i++){//遍历n中模式
for(j=m;j>=0;j--){//体力倒序遍历(01背包优化)
for(k=1;k<=a[i];k++){//遍历收益和体力
if(j>=w[i][k]){
dp[j]=max(dp[j],dp[j-w[i][k]]+v[i][k]);//状态转移方程
}
}
}
}
cout<<dp[m]<<endl;
}
return 0;
}