luogu P1064 金明的预算方案

题目描述

题面 luogu P1064 金明的预算方案

题解

定义数组mw，mv分别储存每个主件的重量及价值，fw，fv分别储存每个主件的附件的重量及价值（按输入顺序确定两个附件的顺序）；
对于当前状态，有四种选择：
1.只选当前的主件
2.选主件与附件1
3.选主件与附件2
4.选主件与附件1 附件2
那么就是一个很明确的dp方程了 套上01背包模板 输出答案f[N].

Code

#include <cstdio>
#include <algorithm>
#define ll long long
using namespace std;
inline int read(){
	int x=0,f=1;char c=getchar();
	while (c<'0'||c>'9') {if (c=='-') f=-1;c=getchar();}
	while (c>='0'&&c<='9') {x=(x<<1)+(x<<3)+(c^48);c=getchar();}
	return x*f;
}
const int maxn=32005;
int n,m;
int mw[maxn],mv[maxn],fw[maxn][3],fv[maxn][3],f[maxn],v,p,q;
inline int max(int a,int b){return a>b?a:b;}
void init(){
	n=read(),m=read();
	for (int i(1);i<=m;++i){
		v=read(),p=read(),q=read();
		if (!q) mw[i]=v,mv[i]=v*p;
		else ++fw[q][0],fw[q][fw[q][0]]=v,fv[q][fw[q][0]]=v*p;
	}
}

signed main(){
	init();
	for (int i(1);i<=m;++i)
		for (int j(n);j>=mw[i];--j){
			f[j]=max(f[j],f[j-mw[i]]+mv[i]);
			if (j>=mw[i]+fw[i][1]) f[j]=max(f[j],f[j-mw[i]-fw[i][1]]+mv[i]+fv[i][1]);
			if (j>=mw[i]+fw[i][2]) f[j]=max(f[j],f[j-mw[i]-fw[i][2]]+mv[i]+fv[i][2]);
			if (j>=mw[i]+fw[i][1]+fw[i][2]) f[j]=max(f[j],f[j-mw[i]-fw[i][1]-fw[i][2]]+mv[i]+fv[i][1]+fv[i][2]);
		}
	printf("%d\n",f[n]);
	return 0; 
}