PAT甲级 1037（C++）

一开始设想，在排序后对coupon从两头逐渐向中间访问，以保证获益最大，后来发现这种想法有bug，原因如下：

若输入

        5

        -5 -4 -3 -2 -1

        5

        -5 1 2 3 4

则输出5，而不是25。

错误代码如下：

#include<iostream>
#include<vector>
#include<algorithm>
#define maxn 100000
using namespace std;
vector<int>coupon;
vector<int>product;
int isvisited[maxn] = { 0 };//product
int isused[maxn] = { 0 };//coupon
int sum=0;
void add(int i) {
	if (isused[i] == 1) return;
	if (coupon[i] < 0) {
		for (int j = 0; j < product.size(); j++) {
			if (product[j] < 0 && isvisited[j] == 0) {
				isvisited[j] = 1; isused[i] = 1;
				sum += coupon[i] * product[j];
				break;
			}
		}
	}
	else if(coupon[i]>0){
		for (int j = product.size() - 1; j >= 0; j--) {
			if (product[j] > 0 && isvisited[j] == 0) {
				isvisited[j] = 1; isused[i] = 1;
				sum += coupon[i] * product[j];
				break;
			}
		}
	}
	isused[i] = 1;
}
int main() {
	int Nc, Np;
	cin>> Nc;
	for (int i = 0; i < Nc; i++) {
		int temp; cin >> temp;
		coupon.push_back(temp);
	}
	cin >> Np;
	for (int i = 0; i < Np; i++) {
		int temp; cin >> temp;
		product.push_back(temp);
	}
	sort(coupon.begin(), coupon.end());
	sort(product.begin(), product.end());
	int i = 0; int j = coupon.size() - 1; 
	while(i<=j){
		add(j); add(i);
		i++; j--;
	}
	cout << sum;
	return 0;
}

因此改用按顺序从头到尾对coupon进行一次遍历，还有一个注意要点，正数要从大到小排序，负数要按绝对值从大到小排序，如：5 4 3 -3 -2 -1。如果仅按最大到小排序，无法求得最大收益。

正确代码如下：

#include<iostream>
#include<vector>
#include<algorithm>
#define maxn 100000
using namespace std;
vector<int>coupon;
vector<int>product;
bool compare(int a, int b) {
	if (a * b > 0)return abs(a) > abs(b);
	else return a > b;
}
int main() {
	int Nc, Np;
	cin>> Nc;
	for (int i = 0; i < Nc; i++) {
		int temp; cin >> temp;
		coupon.push_back(temp);
	}
	cin >> Np;
	for (int i = 0; i < Np; i++) {
		int temp; cin >> temp;
		product.push_back(temp);
	}
	sort(coupon.begin(), coupon.end(),compare);
	sort(product.begin(), product.end(),compare);
	int i = 0; int j = 0; int sum = 0;
	while (i < Nc && j < Np) {
		if (coupon[i] * product[j] > 0) {
			sum += coupon[i] * product[j];
			i++; j++;
		}
		else {
			if (coupon[i] == 0) i++;
			else if (product[j] == 0) j++;
			else if (coupon[i] < 0) j++;
			else i++;
		}
	}
	cout << sum;
	return 0;
}