1705. 吃苹果的最大数目(贪心算法)

1705. 吃苹果的最大数目

难度中等104

有一棵特殊的苹果树，一连 n 天，每天都可以长出若干个苹果。在第 i 天，树上会长出 apples[i] 个苹果，这些苹果将会在 days[i] 天后（也就是说，第 i + days[i] 天时）腐烂，变得无法食用。也可能有那么几天，树上不会长出新的苹果，此时用 apples[i] == 0 且 days[i] == 0 表示。

你打算每天 最多 吃一个苹果来保证营养均衡。注意，你可以在这 n 天之后继续吃苹果。

给你两个长度为 n 的整数数组 days 和 apples ，返回你可以吃掉的苹果的最大数目*。*

示例 1：

输入：apples = [1,2,3,5,2], days = [3,2,1,4,2]
输出：7
解释：你可以吃掉 7 个苹果：
- 第一天，你吃掉第一天长出来的苹果。
- 第二天，你吃掉一个第二天长出来的苹果。
- 第三天，你吃掉一个第二天长出来的苹果。过了这一天，第三天长出来的苹果就已经腐烂了。
- 第四天到第七天，你吃的都是第四天长出来的苹果。

示例 2：

输入：apples = [3,0,0,0,0,2], days = [3,0,0,0,0,2]
输出：5
解释：你可以吃掉 5 个苹果：
- 第一天到第三天，你吃的都是第一天长出来的苹果。
- 第四天和第五天不吃苹果。
- 第六天和第七天，你吃的都是第六天长出来的苹果。

class Solution {
public:
    int eatenApples(vector<int>& apples, vector<int>& days) {
        /* 
            分析：
            每天有apples[i]个新苹果，然后在day[i]天会腐烂，问最多能吃多少苹果
            其实，这个用普通的思维来说，我们每一次都吃快要腐烂的苹果，那么最后肯定能吃到最多的苹果
            map<int,int> mp; // key 腐烂的天数 value 苹果的个数
        
         */

        map<int,int> mp;
        // for(int i=0;i<days.size();i++) {
        //     if(apples[i]==0) {
        //         continue;
        //     }
        //     mp[i+days[i]-1]=pair<int,int>(i,apples[i]);  
        // }
        // for(auto val : mp) {
        //     cout<<val.first<<" "<<val.second.first<<" "<<val.second.second<<endl;
        // }
        int day = 0;
        int ans = 0;
        while(!mp.empty() || day<days.size()) {
            // 开始第day天
            if(day<days.size()  && apples[day]!=0) { // 今天有苹果
                mp[days[day]+day-1]=apples[day];  // 这批苹果最后能保留的天数
            }
            
            if(mp.empty()) {  // 如果没有存储的苹果直接跳过这一天
                day++;
                continue;
            }
            // 有苹果
            auto iter = mp.begin();
            // 开始吃之前的苹果
            iter->second--;
            ans++;
            day++;
            if(day>iter->first || iter->second==0) { // 如果吃完啦，或者过期了就清除苹果
                mp.erase(iter);
            }
        }
        return ans;

    }
};