luogu P1064 金明的预算方案

Posted 2022-11-16 cancers

题目描述

题面 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;