洛谷P1064

[NOIP2006 提高组] 金明的预算方案

题目描述

金明今天很开心，家里购置的新房就要领钥匙了，新房里有一间金明自己专用的很宽敞的房间。更让他高兴的是，妈妈昨天对他说：“你的房间需要购买哪些物品，怎么布置，你说了算，只要不超过 $n$ 元钱就行”。今天一早，金明就开始做预算了，他把想买的物品分为两类：主件与附件，附件是从属于某个主件的，下表就是一些主件与附件的例子：

主件	附件
电脑	打印机，扫描仪
书柜	图书
书桌	台灯，文具
工作椅	无

如果要买归类为附件的物品，必须先买该附件所属的主件。每个主件可以有 $0$ 个、$1$ 个或 $2$ 个附件。每个附件对应一个主件，附件不再有从属于自己的附件。金明想买的东西很多，肯定会超过妈妈限定的 $n$ 元。于是，他把每件物品规定了一个重要度，分为 $5$ 等：用整数 $1\sim 5$ 表示，第 $5$ 等最重要。他还从因特网上查到了每件物品的价格（都是 $10$ 元的整数倍）。他希望在不超过 $n$ 元的前提下，使每件物品的价格与重要度的乘积的总和最大。

设第 $j$ 件物品的价格为 $v_j$，重要度为$w_j$，共选中了 $k$ 件物品，编号依次为 $j_1,j_2,\dots ,j_k$，则所求的总和为：

$v_{j_1}\times w_{j_1}+v_{j_2}\times w_{j_2}+\cdots +v_{j_k}\times w_{j_k}$。

请你帮助金明设计一个满足要求的购物单。

输入格式

第一行有两个整数，分别表示总钱数 $n$ 和希望购买的物品个数 $m$。

第 $2$ 到第 $(m+1)$ 行，每行三个整数，第 $(i+1)$ 行的整数 $v_i$，$p_i$，$q_i$ 分别表示第 $i$ 件物品的价格、重要度以及它对应的的主件。如果 $q_i=0$，表示该物品本身是主件。

输出格式

输出一行一个整数表示答案。

样例 #1

样例输入 #1

1000 5
800 2 0
400 5 1
300 5 1
400 3 0
500 2 0

样例输出 #1

2200

提示

数据规模与约定

对于全部的测试点，保证 $1\le n\le 3.2\times 10^4$，$1\le m\le 60$，$0\le v_i\le 10^4$，$1\le p_i\le 5$，$0\le q_i\le m$，答案不超过 $2\times 10^5$。

NOIP 2006 提高组 第二题

// 买附件就必须买主件。

#include<iostream>
#include<vector>
#include<unordered_map>
using namespace std;

int main(){
    int n, m; // n可以视作其背包容量，m可以视作其物品个数
    cin>>n>>m;
    vector<int> parent(m+1, 0); // 物品从属是谁
    vector<int> money(m+1,0);// 物品的价值
    vector<int> weights(m+1, 0);// 物品的重要度
    unordered_map<int, vector<int>> maps;
    for(int i=1;i<=m;i++){
        cin>>money[i]>>weights[i]>>parent[i];
        maps[parent[i]].emplace_back(i);
    }
    // 要求金钱和重要度最大
    vector<int> dp(n+1, 0); // 这么多钱最多能装多少
    for(int i=1;i<=m;i++){
        for(int j=n;j>=1;j--){
            // 计算买当前是否合适
            if(parent[i] == 0){ // 这时候就需要进行附件的选取
                // 这个可以直接买
                // 这里是买0个附件
                if(j>=money[i]){
                    dp[j] = max(dp[j], dp[j-money[i]]+money[i]*weights[i]);
                }
                // 这里是买1个附件
                auto it = maps[i].size(); // 这是看有几个附件0个，1个 2个附件
                if(it == 0) continue; // 这里就直接返回
                // 这里是存在一个附件，那么就是选不选这个的问题了
                if(j>=money[maps[i][0]]+money[i]){
                    auto mo = money[maps[i][0]]+money[i];
                    auto vo = money[maps[i][0]]*weights[maps[i][0]]+money[i]*weights[i];
                    dp[j] = max(dp[j], dp[j-mo]+vo);
                }
                if(it == 2){
                    // 选1
                    if(j>=money[maps[i][1]]+money[i]){
                        auto mo = money[maps[i][1]]+money[i];
                        auto vo = money[maps[i][1]]*weights[maps[i][1]]+money[i]*weights[i];
                        dp[j] = max(dp[j], dp[j-mo]+vo);
                    }
                    // 选择1和2
                    auto mds = money[maps[i][0]]+money[i]+money[maps[i][1]];
                    auto vds = money[maps[i][0]]*weights[maps[i][0]]+money[i]*weights[i]+money[maps[i][1]]*weights[maps[i][1]];
                    if(j>=mds){
                        dp[j] = max(dp[j], dp[j-mds]+vds);
                    }
                }
            }
        }
    }
    cout<<dp[n]<<endl;
    return 0;
}