luogu P1759 通天之潜水

题目背景

直达通天路·小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);
}