PTA刷题Basic篇——1020.月饼——Day(10)

问题描述

计算出获得指定月饼数量的最大收益。

题目分析

最近leetcode的DFS,回溯,动态规划做多了。感觉这道题有点像进化版的0-1背包问题，我还打算用动态规划，分解成子问题，依次遍历之后找前一个状态的最大值和当前状态的最大值来判断更新。(笑抽)
后来发现这个方法不光复杂而且难以实现，因为如果月饼种类很多的话，这个方法其实是很不稳定的而且前序状态很多很耗时间。所以我们用一个更简单的方法，求平均值法。由于我们知道了每一种月饼的总价值和储存量。所以我们就可以算出每一种月饼的平均价值。然后将这些月饼的平均价值进行排序，利用贪心算法，从最大的价格开始选，使其尽可能满足需求，知道需求量全部被满足。

代码

#include <iostream>
#include <vector>
#include <algorithm>
#include <iomanip>
using namespace std;
struct Mooncake{
    int store;
    int benefit;
    double average;
};
bool cmp(Mooncake m1,Mooncake m2)
{
    return m1.average > m2.average;
}
int main()
{
    int n;//月饼种类
    int D;//需求量
    cin>>n>>D;
    int s;//库存量
    int b;//收益
    vector<Mooncake>m;
    double st[n];
    double bt[n];
    for(int i = 0;i<n;i++)
        cin>>st[i];
    for(int i = 0;i < n;i++)
        cin>>bt[i];
    for(int i = 0;i < n;i++)
    {
       Mooncake moon;
       moon.store = st[i];
       moon.benefit = bt[i];
       moon.average = bt[i] / st[i];
       m.push_back(moon);
    }
    sort(m.begin(),m.end(),cmp);
    double sum = 0;//结果
    for(int i = 0;i < m.size();i++)
    {
        if(m[i].store <= D)
        {
            sum += m[i].store * m[i].average;
            D = D - m[i].store;
        }
        else
        {
            sum += m[i].average * D;
            break;
        }
    }
    cout<<setiosflags(ios::fixed)<<setprecision(2)<<sum;
    return 0;
}

总结

答题用时7min
Q20——finish√