PAT 7-39 魔法优惠券(优先队列的使用)

本题考点:

  • 优先队列的使用

在火星上有个魔法商店,提供魔法优惠券。每个优惠劵上印有一个整数面值K,表示若你在购买某商品时使用这张优惠劵,可以得到K倍该商品价值的回报!该商店还免费赠送一些有价值的商品,但是如果你在领取免费赠品的时候使用面值为正的优惠劵,则必须倒贴给商店K倍该商品价值的金额…… 但是不要紧,还有面值为负的优惠劵可以用!(真是神奇的火星)
例如,给定一组优惠劵,面值分别为1、2、4、-1;对应一组商品,价值为火星币M$7、6、-2、-3,其中负的价值表示该商品是免费赠品。我们可以将优惠劵3用在商品1上,得到M$28的回报;优惠劵2用在商品2上,得到M$12的回报;优惠劵4用在商品4上,得到M$3的回报。但是如果一不小心把优惠劵3用在商品4上,你必须倒贴给商店M$12。同样,当你一不小心把优惠劵4用在商品1上,你必须倒贴给商店M$7。
规定每张优惠券和每件商品都只能最多被使用一次,求你可以得到的最大回报。

本题是优先队列的使用经典案例,我们将正数和负数分开存放,正数放到大顶堆中,负数放到小顶堆中,然后分别从两个堆中去除对应的值相乘然后相加即可获得最大值。

完整代码如下:

#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
using namespace std;

priority_queue<int> posiGoods, posiDiscounts;   // 正数采用大顶堆
priority_queue<int, vector<int>, greater<int>> negaGoods, negaDiscounts;    // 负数采用小顶堆

int main()
{
    int n, temp;
    scanf("%d", &n);
    for (int i = 0; i < n; i++)
    {
        scanf("%d", &temp);
        if (temp > 0)
            posiGoods.push(temp);
        else
            negaGoods.push(temp);
    }
    scanf("%d", &n);
    for (int i = 0; i < n; i++)
    {
        scanf("%d", &temp);
        if (temp > 0)
            posiDiscounts.push(temp);
        else
            negaDiscounts.push(temp);
    }
    int sum = 0;
    while (!posiGoods.empty() && !posiDiscounts.empty())
    {
        sum += posiGoods.top() * posiDiscounts.top();
        posiDiscounts.pop();
        posiGoods.pop();
    }
    while (!negaGoods.empty() && !negaDiscounts.empty())
    {
        sum += negaGoods.top() * negaDiscounts.top();
        negaGoods.pop();
        negaDiscounts.pop();
    }
    printf("%d", sum);
    return 0;
}
posted @ 2020-04-10 13:21  南风sa  阅读(244)  评论(0编辑  收藏  举报