洛谷P1064 金明的预算方案（DP）

题意分析

变形背包
对于每种商品有5种决策
不买
买主件
买主件+1号附件
买主件+2号附件
买主件+1号和2号附件
对于5种决策分别DP即可

代码总览

#include<bits/stdc++.h>
using namespace std;
const int nmax = 35000;
const int mmax = 100;
int mainv[mmax],mainp[mmax];
int attv[mmax][3],attp[mmax][3];
int dp[nmax];
int tot,n,cnt;
int main(){
    while(scanf("%d %d",&tot,&n) != EOF){
        cnt = 0;
        for(int i = 1;i<=n;++i){
            int v,q,p;
            scanf("%d %d %d",&v,&p,&q);
            if(q == 0){
                mainv[i] = v;
                mainp[i] = v*p;
            }else{
                attv[q][0] ++;
                attv[q][attv[q][0]] = v;
                attp[q][attv[q][0]] = v * p;
            }
        }
        for(int i =1; i<=n;++i){
            for(int j = tot;mainv[i] !=0&& j>=mainv[i];--j){
                if(j>=mainv[i]) dp[j] = max(dp[j],dp[j-mainv[i]] + mainp[i]);
                if(j>=mainv[i] + attv[i][1]) dp[j] = max(dp[j],dp[j - mainv[i] - attv[i][1]] + mainp[i] + attp[i][1] );
                if(j>=mainv[i] + attv[i][2]) dp[j] = max(dp[j],dp[j - mainv[i] - attv[i][2]] + mainp[i] + attp[i][2] );
                if(j>=mainv[i] + attv[i][1] + attv[i][2])
                    dp[j] = max(dp[j], dp[j- mainv[i] - attv[i][1] - attv[i][2]] + mainp[i] + attp[i][1] + attp[i][2]);

            }
        }
        int ans = -0x3f3f3f3f;
        for(int i = 0;i<=tot;++i) ans = max(dp[i],ans);
        printf("%d\n",ans);

    }
    return 0;
}