PAT 甲级 1037 Magic Coupon (25 分)

题目描述

The magic shop in Mars is offering some magic coupons. Each coupon has an integer N printed on it, meaning that when you use this coupon with a product, you may get N times the value of that product back! What is more, the shop also offers some bonus product for free. However, if you apply a coupon with a positive N to this bonus product, you will have to pay the shop N times the value of the bonus product… but hey, magically, they have some coupons with negative N’s!
For example, given a set of coupons { 1 2 4 −1 }, and a set of product values { 7 6 −2 −3 } (in Mars dollars M$) where a negative value corresponds to a bonus product. You can apply coupon 3 (with N being 4) to product 1 (with value M$7) to get M$28 back; coupon 2 to product 2 to get M$12 back; and coupon 4 to product 4 to get M$3 back. On the other hand, if you apply coupon 3 to product 4, you will have to pay M$12 to the shop.
Each coupon and each product may be selected at most once. Your task is to get as much money back as possible.

输入

Each input file contains one test case. For each case, the first line contains the number of coupons $N_c$ , followed by a line with $N_c$ coupon integers. Then the next line contains the number of products $N_P$ , followed by a line with $N_P$ product values. Here $1\le N_c,N_P\le 10^5$, and it is guaranteed that all the numbers will not exceed $2^{30}$.

输出

For each test case, simply print in a line the maximum amount of money you can get back.

思路

题目描写一大堆其实就是两个数组选取一部分的值相乘再相加求最大值。
做法就是先排序，可以把正数负数分开，正数乘正数，负*负，相加即可。

代码

#include<iostream>
#include<cstdio>
#include<stdlib.h>
#include<algorithm>
using namespace std;
int P_z[100005], P_f[100005],C_z[100005],C_f[100005];
int npz=0, npf=0, ncz=0, ncf=0;
long long ans = 0;
bool cmp(int a, int b)
{
	return a > b;
}
int main()
{
	int Nc, Np;
	cin >> Nc;
	for (int i = 0; i < Nc; i++)
	{
		int a;
		cin >> a;
		if (a >= 0)
		{
			C_z[ncz++] = a;
		}
		else
		{
			C_f[ncf++] = -a;
		}
	}
	cin >> Np;
	for (int i = 0; i < Np; i++)
	{
		int a;
		cin >> a;
		if (a >= 0)
		{
			P_z[npz++] = a;
		}
		else
		{
			P_f[npf++] = -a;
		}
	}
	sort(C_z, C_z + ncz, cmp);
	sort(C_f, C_f + ncf, cmp);
	sort(P_z, P_z + npz, cmp);
	sort(P_f, P_f + npf, cmp);
	int nz = ncz;
	if (nz > npz)
	{
		nz = npz;
	}
	for (int i = 0; i < nz; i++)
	{
		ans += C_z[i] * P_z[i];
	}
	int nf = ncf;
	if (nf > npf)
	{
		nf = npf;
	}
	for (int i = 0; i < nf; i++)
	{
		ans += C_f[i] * P_f[i];
	}
	printf("%lld\n", ans);
}