中级-购物单

题目

王强今天很开心，公司发给N元的年终奖。王强决定把年终奖用于购物，他把想买的物品分为两类：主件与附件，附件是从属于某个主件的，下表就是一些主件与附件的例子：
主件 附件
电脑 打印机，扫描仪
书柜 图书
书桌 台灯，文具
工作椅 无
如果要买归类为附件的物品，必须先买该附件所属的主件。每个主件可以有 0 个、 1 个或 2 个附件。附件不再有从属于自己的附件。王强想买的东西很多，为了不超出预算，他把每件物品规定了一个重要度，分为 5 等：用整数 1 ~ 5 表示，第 5 等最重要。他还从因特网上查到了每件物品的价格（都是 10 元的整数倍）。他希望在不超过 N 元（可以等于 N 元）的前提下，使每件物品的价格与重要度的乘积的总和最大。
设第 j 件物品的价格为 v[j] ，重要度为 w[j] ，共选中了 k 件物品，编号依次为 j 1 ， j 2 ，……， j k ，则所求的总和为：
v[j 1 ]w[j 1 ]+v[j 2 ]*w[j 2 ]+ … +v[j k ]*w[j k ] 。（其中 为乘号）
请你帮助王强设计一个满足要求的购物单。

输入：

输入的第 1 行，为两个正整数，用一个空格隔开：N m
（其中 N （ <32000 ）表示总钱数， m （ <60 ）为希望购买物品的个数。）

从第 2 行到第 m+1 行，第 j 行给出了编号为 j-1 的物品的基本数据，每行有 3 个非负整数 v p q

（其中 v 表示该物品的价格（ v<10000 ）， p 表示该物品的重要度（ 1 ~ 5 ）， q 表示该物品是主件还是附件。如果 q=0 ，表示该物品为主件，如果 q>0 ，表示该物品为附件， q 是所属主件的编号）

输出：

输出文件只有一个正整数，为不超过总钱数的物品的价格与重要度乘积的总和的最大值（ <200000 ）。

样例输入：

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

样例输出：

2200

解题思路：

本题使用贪心算法，先对初始数据按价格进行排序，为了方便找到附件属于哪个主件，在输入的时候，拷贝一份副本raw_h。从排序后的，先选价格最高的，判断是否超过奖金，超过了就不能购买该商品，选取下一个价格最高的商品，否则，判断是否是附件，如果是附件，则加上主件价格，看是否超过奖金，如果超过了，则需要将目前购买价格回退。如果没有，则判断是否是最大的总和。基本思路就是如此，采用贪心算法解决，具体的代码注释有空再补吧。

#include <iostream>
#include <stdio.h>
#include <algorithm>
using namespace std;

struct huff
{
    int v;
    int p;
    int q;
};

int cmp(huff a, huff b)
{
    return a.v > b.v;
}

bool existed(int* a, int b, int len)
{
    if (len <= 0) {
        return false;
    }
    for (int i = 0; i < len; ++i) {
        if(a[i] == b)
            return true;
    }
    return false;
}

int main()
{
    int N,m;
    scanf("%d %d",&N,&m);
    huff h[65];
    huff raw_h[65];
    for (int i = 0; i < m; ++i) {
        scanf("%d %d %d",&h[i].v,&h[i].p,&h[i].q);
        raw_h[i] = h[i];
    }

    sort(h, h + m, cmp);

    //for (int i = 0; i < m; ++i) {
    //    printf("%d %d %d\n",raw_h[i].v,raw_h[i].p,raw_h[i].q);
    //    printf("sort -- %d %d %d\n",h[i].v,h[i].p,h[i].q);
    //}

    int sum = 0;
    int mul = 0;
    int max = 0;
    int ex[65];
    int len = 0;
    for (int i = 0; i < m; ++i) {
        memset(ex, 0, 65*sizeof(int));
        len = 0;
        sum = h[i].v;
        mul = h[i].v * h[i].p;

        if (sum > N) {
            continue;
        }
        //如果不是主件
        if (h[i].q != 0) {
            sum += raw_h[h[i].q - 1].v;
            if (sum > N) {
                sum -= raw_h[h[i].q - 1].v;
                continue;
            }
            ex[len++] = h[i].q;
            mul += raw_h[h[i].q - 1].v * raw_h[h[i].q - 1].p;
            if (sum <= N) {
                if (mul >= max) {
                    max = mul;
                }
            }
        }
        if (sum <= N) {
            if (mul >= max) {
                max = mul;
            }
        }

        //贪心下一个最贵的
        for (int j = i + 1; j < m && !existed(ex, j, len); ++j) {
            sum += h[j].v;

            if (sum > N) {
                sum -= h[j].v;
                continue;
            }
            //cout<<"j "<< j<<"sum "<<sum<<endl;
            //如果不是主件
            if (h[j].q != 0) {
                sum += raw_h[h[j].q - 1].v;
                if (sum > N) {
                    sum -= h[j].v;
                    sum -= raw_h[h[j].q - 1].v;
                    continue;
                }
                ex[len++] = h[j].q;
                mul += raw_h[h[j].q - 1].v * raw_h[h[j].q - 1].p;
                if (sum <= N) {
                    if (mul >= max) {
                        max = mul;
                    }
                }
            }
            mul += h[j].v * h[j].p;
            if (sum <= N) {
                if (mul >= max) {
                    max = mul;
                }
            }
        }
        //cout<<"The "<<i<<" "<<max<<endl;
    }
    cout<<max<<endl;

    return 0;
}