7-39 魔法优惠券 (25 分)
在火星上有个魔法商店,提供魔法优惠券。每个优惠劵上印有一个整数面值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。
规定每张优惠券和每件商品都只能最多被使用一次,求你可以得到的最大回报。
输入格式:
输入有两行。第一行首先给出优惠劵的个数N,随后给出N个优惠劵的整数面值。第二行首先给出商品的个数M,随后给出M个商品的整数价值。N和M在[1, 10
6
]之间,所有的数据大小不超过2
30
,数字间以空格分隔。
输出格式:
输出可以得到的最大回报。
输入样例:
4 1 2 4 -1
4 7 6 -2 -3
输出样例:
43
思路分析:
将正数与负数分开,按从大到小顺序排列,然后一一匹配。
#include <iostream>
#include <queue>
using namespace std;
int main()
{
priority_queue<int> Q1, Q1_1, Q2, Q2_2;
int N, M, tmp;
long long ans = 0;
cin>>N;
while(N--)
{
cin>>tmp;
if(tmp >= 0)
Q1.push(tmp);
else
{
tmp *= -1;
Q1_1.push(tmp);
}
}
cin>>M;
while(M--)
{
cin>>tmp;
if(tmp >= 0)
Q2.push(tmp);
else
{
tmp *= -1;
Q2_2.push(tmp);
}
}
while(!Q1.empty() && !Q2.empty())
{
int q1 = Q1.top();
int q2 = Q2.top();
if(q1*q2 > 0) ans += q1*q2;
Q1.pop();
Q2.pop();
}
while(!Q1_1.empty() && !Q2_2.empty())
{
int q1 = Q1_1.top();
int q2 = Q2_2.top();
if(q1*q2 > 0) ans += q1*q2;
Q1_1.pop();
Q2_2.pop();
}
cout<<ans<<endl;
return 0;
}