PAT B1020 月餅（簡單貪心）

描述

月餅是中國人在中秋佳節時吃的一種傳統食品，不同地區有許多不同風味的月餅。現給定所有種類月餅的庫存量、總售價、以及市場的最大需求量，請你計算可以獲得的最大收益是多少。

注意：銷售時允許取出一部分庫存。樣例給出的情形是這樣的：假如我們有 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

Solution

按照單價貪心。注意總價、儲存量的型別，迴圈判斷儘量用for，否則很容易忘記迴圈終止條件。

#include <iostream>
#include <algorithm>
#include <cstdio>
using namespace std;
const int maxn = 1001;
struct Pie
{
    double store_num, total_price, price;
    bool operator<(const Pie &t) const
    {
        return price > t.price;
    }
};
Pie pie[maxn];
int main()
{
    // freopen("in.txt", "r", stdin);
    int N;
    int max_need;
    scanf("%d", &N);
    scanf("%d", &max_need);
    for (int i = 0; i < N; i++)
        scanf("%lf", &pie[i].store_num);
    for (int i = 0; i < N; i++)
        scanf("%lf", &pie[i].total_price);
    for (int i = 0; i < N; i++)
        pie[i].price = pie[i].total_price / pie[i].store_num;
    sort(pie, pie + N);
    double ans_price = 0;
    // 能用for迴圈儘量不用while，在這裡如果用while不限定start<N條件很容易會出錯
    for (int start = 0; start < N && max_need != 0; start++)
    {
        if (max_need - pie[start].store_num >= 0)
        {
            max_need -= pie[start].store_num;
            ans_price += pie[start].total_price;
        }
        else
        {
            ans_price += pie[start].price * max_need;
            max_need = 0;
        }
    }
    printf("%.2lf\n", ans_price);
    return 0;
}