题目概述
题目:https://vjudge.net/problem/HDU-2602
这里借用一个魔改翻译:
已知N个糖果的重量和价值. 我们有一个口袋, 最多可以装V重量的糖果. 问口袋最多能放多少价值的糖果进去?
输入描述:
输入的第一行是T, 表示有T组数据.
每组数据由三行组成.
第一行包含两个整数N和V(N <= 1000, V <= 1000). N表示糖果的个数, V表示口袋的载重.
第二行包含N个整数, 表示每一颗糖果的价值.
第三行包含N个整数, 表示每一颗糖果的重量.
输出描述:
对每一组数据, 输出口袋最终可以放进去糖果的价值.
样例输入:
1
5 10
1 2 3 4 5
5 4 3 2 1
样例输出:
14
思路分析
0/1背包是dp的经典问题。对于dp,我们的思路还是:定义状态,再找到状态转移方程,然后得出解。
首先,什么是状态?状态指的应该是:在放入前i个物品到容量为j的背包中获得的最大价值。现在,一步一步地填充我们的表:
第一行i=0,不选择糖果,全为0。
第二行开始,0<=j<=4范围内,因为放不下第一个糖果(体积为5),所以为0,后面几个最大的就是放入第一个糖果。
第三行,0<=j<=3范围内也为0(同上),j=4开始,可以放下第二个糖果,故为2.
j=5,需要选择是否放入第二个糖果:
若不放入,则dp[i][j]=dp[i-1][j]。
若放入,则放入后i-1=1,j-vi=1,就相当于(dp[1][1]+pi).
……
由上述推导过程,我们可以得到状态转移方程
根据这个我们就能算出整个状态转移矩阵,然后输出对应格子就行了。
完整代码
#include <iostream>
#include <cstring>
using namespace std;
int dp[1011][1011]={0};
struct candy{
int p;
int v;
}candys[1011];
int ans(int n,int v)
{
memset(dp,0,sizeof(dp));
for(int i=1;i<=n;i++)
{
for(int j=0;j<=v;j++)
{
if(j<candys[i].v)
{
dp[i][j]=dp[i-1][j];
}
else
{
dp[i][j]=max(dp[i-1][j],dp[i-1][j-candys[i].v]+candys[i].p);
}
}
}
return dp[n][v];
}
int main()
{
int t;
cin>>t;
while(t--)
{
int n,v;
cin>>n>>v;
for(int i=1;i<=n;i++) cin>>candys[i].p;
for(int i=1;i<=n;i++) cin>>candys[i].v;
cout<<ans(n,v)<<endl;
}
return 0;
}