原题:https://pintia.cn/problem-sets/994805260223102976/problems/994805301562163200
月饼是中国人在中秋佳节时吃的一种传统食品,不同地区有许多不同风味的月饼。现给定所有种类月饼的库存量、总售价、以及市场的最大需求量,请你计算可以获得的最大收益是多少。
注意:销售时允许取出一部分库存。样例给出的情形是这样的:假如我们有 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
题目分析:
本题目类似于连续背包问题,可以使用贪心算法解决,并且能找到最优解。
⾸先根据⽉饼的总价和数量计算出每⼀种⽉饼的单价,然后将⽉饼数组按照单价从⼤到⼩排序,根据需求量D的⼤⼩,从单价最⼤的⽉饼开始售卖,将销售掉这种⽉饼的价格累加到result中,再相应减少需求量直到0,最后输出result即可。
#include<iostream>
#include<algorithm>
using namespace std;
//除了市场需求是整数
int main()
{
int N;
int D;
cin >> N >> D;
double* stock = new double[N];//库存量
double* price = new double[N];//月饼单价
double* density = new double[N]; //月饼的价值密度
int i;
for (i = 0; i < N; i++) cin >> stock[i];
for (i = 0; i < N; i++) {
cin >> price[i];
density[i] = price[i] / stock[i];
}
double profit = 0; //总收益
while (D) {//分配需求量
double* max = max_element(density, density + N);
int index = max - density; //最大价值密度的位置
double allocation = stock[index] < D ? stock[index] : D;
D -= allocation; //需求量降低
profit += allocation*density[index]; //计算已得利益
density[index] = 0; //已经分配的
}
printf("%.2f", profit);
cout << endl;
delete[] stock, price, density;
system("pause");
return 0;
}
注意,我在做题的时候发现一个测试点始终过不了,最后检查是由于:
本题的信息只能得出N和D是正整数,其他的应该声明为浮点数
,不应该受参考样例的迷惑。
当然,寻找最大值的代码也可以自己写,这里,我是用的是<algorithm>
里的max_element()
函数
风格二:
来自柳婼 の blog
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
struct mooncake {
float mount, price, unit;
};
int cmp(mooncake a, mooncake b) {
return a.unit > b.unit;
}
int main() {
int n, need;
cin >> n >> need;
vector<mooncake> a(n);
for (int i = 0; i < n; i++) scanf("%f", &a[i].mount);
for (int i = 0; i < n; i++) scanf("%f", &a[i].price);
for (int i = 0; i < n; i++) a[i].unit = a[i].price / a[i].mount;
sort(a.begin(), a.end(), cmp);
float result = 0.0;
for (int i = 0; i < n; i++) {
if (a[i].mount <= need) {
result = result + a[i].price;
}
else {
result = result + a[i].unit * need;
break;
}
need = need - a[i].mount;
}
printf("%.2f", result);
return 0;
}
使用了STL中
<algorithm>
的sort()
排序函数。
推荐阅读:
1.STL中vector容器
https://blog.csdn.net/Africa_South/article/details/87875863
2.STL中排序函数sort()
https://blog.csdn.net/Africa_South/article/details/87905127