PAT(天梯赛练习题) -L2-003. 月饼(贪心)

时间:2023-02-23 11:03:21

题目链接

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

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

输入格式:

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

输出格式:

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

输入样例:
3 20
18 15 10
75 72 45
输出样例:
94.50

明显的贪心,先算出每种月饼的单份售价,用结构体排序。有个坑点是会卡精度所以读入时就用double吧。


代码如下

#include<iostream>
#include<string>
#include<cstring>
#include<vector>
#include<map>
#include<algorithm>
#include<queue>
#include<set>
#include<cstdio>
#include<functional>
#include<iomanip>
#include<cmath>
#include<stack>
using namespace std;
typedef long long LL;
typedef unsigned long long ull;
struct nodes {
  double num, tot, di;
  bool operator <(const nodes &a)const {
    return di > a.di;
  }
}a[11000];
int main() {
  //freopen("D:\\cpp\\test.txt", "r", stdin);  
  int n, m;
  scanf("%d%d", &n, &m);
  for (int i = 0; i < n; i++)
    scanf("%lf", &a[i].num);
  for (int i = 0; i < n; i++)
    scanf("%lf", &a[i].tot);
  for (int i = 0; i < n; i++)
    a[i].di = a[i].tot / a[i].num;
  sort(a, a + n);
  double ans = 0.0;
  for (int i = 0; i < n; i++) {
    if (m > a[i].num) {
      ans += a[i].tot;
      m -= a[i].num;
    }
    else {
      ans += a[i].di*double(m);
      m = 0;
      break;
    }
  }
  printf("%.2lf\n", ans);
  return 0;
}