PTA——7-39 魔法优惠券 (25分)

PTA——7-39 魔法优惠券 (25分)

链接：https://pintia.cn/problem-sets/15/problems/866?redir_from_old=true

在火星上有个魔法商店，提供魔法优惠券。每个优惠劵上印有一个整数面值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 <stdio.h>
#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
bool cmp( int a,int b){ 
	return a<b;
}
bool cmp1( int a,int b){ 
	return a>b;
}
int main() {
	int n,m;
	while( scanf( "%d",&n)!=EOF) {
		long long sum=0;
		int a[n+5];
		for( int i=0; i<n; i++) {
			scanf( "%d",&a[i]);
		}
		scanf( "%d",&m);
		int b[m+5];
		for( int i=0; i<m; i++) {
			int k;
			scanf( "%d",&b[i]);
		}
		sort( a,a+n,cmp);
		sort( b,b+m,cmp);
		int t=n<m?n:m;
		for( int i=0; i<t; i++) {
			if(a[i]<0&&b[i]<0){
				sum+=(a[i]*b[i]);
			}
		}
		sort( a,a+n,cmp1);
		sort( b,b+m,cmp1);
		for( int i=0; i<t; i++) {
			if(a[i]>0&&b[i]>0){
				sum+=(a[i]*b[i]);
			}
		}
		printf("%lld\n",sum);
	}
	return 0;

}

正确代码2：

不知道为什么，数组设为局部变量就ac不了

#include <stdio.h>
#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
bool cmp( int a,int b){ 
	return abs( a)>abs( b);
}
int a1[1000000],a2[1000000];
int b1[1000000],b2[1000000];
int main() {
	int n,m;
	while( scanf( "%d",&n)!=EOF) {
		
		int z1=0,f1=0,z2=0,f2=0;
		long long sum=0;
		for( int i=0; i<n; i++) {
			int k;
			scanf( "%d",&k);
			if( k<0) {
				a2[f1++]=k;
			}
			else if( k>0) {
				a1[z1++]=k;
			}
		}
		scanf( "%d",&m);
		
		for( int i=0; i<m; i++) {
			int k;
			scanf( "%d",&k);
			if( k<0) {
				b2[f2++]=k;
			}
			else if( k>0){
				b1[z2++]=k;
			}
		}
		sort( a1,a1+z1,cmp);
		sort( a2,a2+f1,cmp);
		sort( b1,b1+z2,cmp);
		sort( b2,b2+f2,cmp);
		int lenz=z1<z2?z1:z2;
		int lenf=f1<f2?f1:f2;
		for( int i=0; i<lenz; i++) {
			sum+=( a1[i]*b1[i]);
			//printf( "%d %d\n",a1[i],b1[i]);
		}
		for( int i=0; i<lenf; i++) {
			sum+=( a2[i]*b2[i]);
			//printf( "%d %d\n",a2[i],b2[i]);
		}
		printf("%lld\n",sum);
	}
	return 0;

}