PTA数据结构与算法分析 7-39 魔法优惠券 (25 分)

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

思路:用c++的sort,老实说这几题用sort用到内心愧疚,不练习排序算法真的可以吗?
第一次下写就是单纯的排序,然后商品与优惠卷的数组顺序相乘就行了,然后倒数第二个测试点错误:

5 4 2 1 0 -1
4 7 6 -2 -3 1
这组数据就会出错,结果是40,因此在乘积为负数时,谁的值为正谁的下标就往下一个移动,另一个下标也要移动,还要注意下标不要越界的问题
代码如下:

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
int coupon[1000010],cdy[1000010];
bool cmp(int a,int b)
{
 if(a > b)
  return true;
 return false;
}
int main()
{
// freopen("D://in.txt","r",stdin);
 int N,M,sum = 0;
 cin >> N;
 for(int i=0;i < N;i++){
  scanf("%d",&coupon[i]);
 }
 cin >> M;
 for(int i=0;i < M;i++){
  scanf("%d",&cdy[i]);
 }
 sort(coupon,coupon+N,cmp);
 sort(cdy,cdy+M,cmp);
 for(int i=0,ptr=0;i < N && ptr < M;i++,ptr++){
  if(coupon[i] * cdy[ptr] >= 0){
   sum += coupon[i] * cdy[ptr];
  }
  else if(coupon[i] > 0){
   while(coupon[i+1] >= 0 && i+1 < N){
    i++;
    if(ptr + 2 < M)
     ptr++;
   }
  }
  else if(cdy[i] > 0){
   while(coupon[ptr+1] >= 0){
    ptr++;
    if(i + 2 < N)
     i++;
   }
  }
 }
 printf("%d",sum);
 //fclose(stdin);
 return 0;
}```

猜你喜欢

转载自blog.csdn.net/qq_43189757/article/details/88756909