题目背景

直达通天路·小A历险记第三篇

题目描述

在猴王的帮助下,小A终于走出了这篇荒山,却发现一条波涛汹涌的河拦在了自己的面前。河面上并没有船,但好在小A有n个潜水工具。由于他还要背重重的背包,所以他只能背m重的工具,又因为他的力气并不是无限的,河却很宽,所以他只能背有v阻力的工具。但是这条河下有非常重要的数据,所以他希望能够停留的时间最久。于是他找到了你,让你告诉他方案。

输入格式

三个数m,v,n如题目所说

接下来n行,每行三个数ai,bi,ci分别表示所含的重力,阻力,能够支撑的时间

输出格式

第一行一个数,表示最长的时间

接下来一行,若干个数,表示所选的物品

说明/提示

1<=m,v<=200,n<=100

数据保证一定有方案。

若有多种方案,输出前面尽量小的方案。


01背包附带两个限制条件

除此之外还要一点递归技巧来输出方案

让 f[i][j][k] 表示dp[i][j]转移时是否选择了k

#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define int long long
using namespace std;
const int N=205;
int a[N],b[N],val[N];
int dp[N][N];
bool f[N][N][N];
void output(int i,int j,int k){
if(i==1){
if(f[i][j][k])printf("1 ");
return;
}
if(f[i][j][k])output(i-1,j-a[i],k-b[i]);
else output(i-1,j,k);
if(f[i][j][k])printf("%d ",i);
}
signed main(){
int m,v,n;
cin>>m>>v>>n;
for(int i=1;i<=n;i++)
scanf("%lld%lld%lld",&a[i],&b[i],&val[i]);
for(int i=1;i<=n;i++){
for(int j=m;j>=a[i];j--)
for(int k=v;k>=b[i];k--)
if(dp[j][k]<dp[j-a[i]][k-b[i]]+val[i]){
dp[j][k]=dp[j-a[i]][k-b[i]]+val[i];
f[i][j][k]=1;
}
}
cout<<dp[m][v]<<endl;
output(n,m,v);
}
05-23 23:16