#Jin Ge Jin Qu hao UVA - 12563
【分析】
本题是0-1背包问题的扩充。本题包括两个要点,一个要点是完全套用0-1背包问题的解法求得最大歌曲数目,另一个要点(难点)是求当取得最大歌曲数目时所对应的最长时间。下面代码使用了len[i][j]表示用前i个物品装入容量为j的背包可以获得的最长时间,特别注意len的递推一定要实时和dp的递推保持一致(最长时间是在最多歌曲的基础上进行的,这是解决本问题的关键点)。
#include<iostream>
#include<cstring>
using namespace std;
int T;
int n,t;
int w[55];
int dp[55][10000];
int len[55][10000];
int kase=0;
int main()
{
cin>>T;
while(T--){
cin>>n>>t;
for(int i=1;i<=n;i++){
cin>>w[i];
}
memset(dp,0,sizeof(dp));
memset(len,0,sizeof(len));
for(int i=1;i<=n;i++){
for(int j=t;j>=1;j--){
if(j>w[i]){
dp[i][j]=max(dp[i-1][j],dp[i-1][j-w[i]]+1);
if(dp[i-1][j]>dp[i-1][j-w[i]]+1){
len[i][j]=len[i-1][j];
}
else if(dp[i-1][j]<dp[i-1][j-w[i]]+1){
len[i][j]=len[i-1][j-w[i]]+w[i];
}
else{
len[i][j]=max(len[i-1][j],len[i-1][j-w[i]]+w[i]);
}
}
else{
dp[i][j]=dp[i-1][j];
len[i][j]=len[i-1][j];
}
}
}
cout<<"Case "<<++kase<<": "<<dp[n][t]+1<<" "<<len[n][t]+678<<endl;
}
return 0;
}
当然也可以根据0-1背包合并if-else代码来稍微简化下代码如下:
#include<iostream>
#include<cstring>
using namespace std;
int T;
int n,t;
int w[55];
int dp[55][10000];
int len[55][10000];
int kase=0;
int main()
{
cin>>T;
while(T--){
cin>>n>>t;
for(int i=1;i<=n;i++){
cin>>w[i];
}
memset(dp,0,sizeof(dp));
memset(len,0,sizeof(len));
for(int i=1;i<=n;i++){
for(int j=t;j>=1;j--){
dp[i][j]=dp[i-1][j];
len[i][j]=len[i-1][j];
if(j>w[i]){
if(dp[i][j]<dp[i-1][j-w[i]]+1){
len[i][j]=len[i-1][j-w[i]]+w[i];
dp[i][j]=dp[i-1][j-w[i]]+1;
}
else if(dp[i][j]==dp[i-1][j-w[i]]+1){
len[i][j]=max(len[i-1][j],len[i-1][j-w[i]]+w[i]);
}
}
}
}
cout<<"Case "<<++kase<<": "<<dp[n][t]+1<<" "<<len[n][t]+678<<endl;
}
return 0;
}