PTA 1020 C++实现 月饼(注意测试点2,3)

题目

1020 月饼 (25分)

月饼是中国人在中秋佳节时吃的一种传统食品，不同地区有许多不同风味的月饼。现给定所有种类月饼的库存量、总售价、以及市场的最大需求量，请你计算可以获得的最大收益是多少。

注意：销售时允许取出一部分库存。样例给出的情形是这样的：假如我们有 3 种月饼，其库存量分别为 18、15、10 万吨，总售价分别为 75、72、45 亿元。如果市场的最大需求量只有 20 万吨，那么我们最大收益策略应该是卖出全部 15 万吨第 2 种月饼、以及 5 万吨第 3 种月饼，获得 72 + 45/2 = 94.5（亿元）。

输入格式：
每个输入包含一个测试用例。每个测试用例先给出一个不超过 1000 的正整数 N 表示月饼的种类数、以及不超过 500（以万吨为单位）的正整数 D 表示市场最大需求量。随后一行给出 N 个正数表示每种月饼的库存量（以万吨为单位）；最后一行给出 N 个正数表示每种月饼的总售价（以亿元为单位）。数字间以空格分隔。

输出格式：
对每组测试用例，在一行中输出最大收益，以亿元为单位并精确到小数点后 2 位。

输入样例：

3 20
18 15 10
75 72 45

输出样例：

94.50

分析

把月饼按性价比从高到低排序。
找出性价比最高(单价最高)的月饼种类，如果其库存小于需求，则这种月饼全部抛售。如果还有需求则开始售卖性价比第二高的。依次类推。
测试点2要注意：月饼的库存不一定是整数，要用浮点数来储存。
测试点3要注意：需求D大于所有月饼库存。即供不应求。如果考虑不充分可能导致段错误(越界等)或数值错误。

我用的sort()函数来进行排序。

代码

#include<iostream>
#include<vector>
#include<algorithm>
#include<iomanip>
using namespace std;
struct goods {
	double amount;   //库存
	double z;		//总价
	double price;//单价
	goods():amount(0),z(0),price(0){}
	bool operator >(goods& a) { return price > a.price; }
};
bool cmp(goods a, goods b)
{
	return a>b;
}
int main()
{
	vector<goods> v;
	vector<double>z, a;    //总价和库存
	int n, d;
	double temp;
	goods t;
	cin >> n >> d;
	for (int i = 0; i < n; i++) {
		cin >>temp;
		a.push_back(temp);
	}
	for (int i = 0; i < n; i++) {
		cin >> temp;
		z.push_back(temp);
	}
	for (int i = 0; i < n; i++) {
		t.amount=a[i];
		t.z = z[i];
		t.price = (double)t.z / t.amount;
		v.push_back(t);
	}
	sort(v.begin(), v.end(), cmp);
	double sum=0;
	double s = 0;
	int i = 0;
	while (s != d&&i<(int)v.size()) {
		if (v[i].amount + s < d) {
			sum += v[i].z;
			s += v[i].amount;
			++i;
		}
		else {
			sum += v[i].price*(d-s);
			s = d;
		}
	}
	cout << fixed << setprecision(2)<<sum;
	return 0;
}