题目
有一棵特殊的苹果树,一连 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 个苹果: - 第一天到第三天,你吃的都是第一天长出来的苹果。 - 第四天和第五天不吃苹果。 - 第六天和第七天,你吃的都是第六天长出来的苹果。
解题
方法一:贪心+优先队列
用优先队列存储,vector{苹果到期日子,苹果个数}
先摘当天的苹果
然后把过期的苹果扔掉
每次吃最先要过期的苹果。(先pop出来,苹果个数减1,如果个数大于0或者第二天还能吃,就push回来)
class Solution { public: int eatenApples(vector<int>& apples, vector<int>& days) { struct cmp{ bool operator()(vector<int>&a,vector<int>&b){ return a[0]>b[0]; } }; priority_queue<vector<int>,vector<vector<int>>,cmp> q; int n=apples.size(); int time=0,res=0;//time记录日期,res记录吃的苹果数 while(time<n||!q.empty()){ if(time<n&&apples[time]>0) q.push({time+days[time]-1,apples[time]}); while(!q.empty()&&time>q.top()[0]) q.pop();//将过期苹果扔掉 if(!q.empty()){//如果还有苹果,就吃掉,将它pop出来,吃掉一个,如果数量还>0的就push回去 vector<int> cur=q.top(); q.pop(); if(--cur[1]>0&&cur[0]>time) q.push(cur);//数量大于0且第二天还能吃,就push回去 res++; } time++; } return res; } };
可以发现把vector
换成pair
后具有更快的速度
class Solution { public: int eatenApples(vector<int>& apples, vector<int>& days) { struct cmp{ bool operator()(pair<int,int>&a,pair<int,int>&b){ return a.first>b.first; } }; priority_queue<pair<int,int>,vector<pair<int,int>>,cmp> q; int n=apples.size(); int time=0,res=0;//time记录日期,res记录吃的苹果数 while(time<n||!q.empty()){ if(time<n&&apples[time]>0) q.push({time+days[time]-1,apples[time]}); while(!q.empty()&&time>q.top().first) q.pop();//将过期苹果扔掉 if(!q.empty()){//如果还有苹果,就吃掉,将它pop出来,吃掉一个,如果数量还>0的就push回去 pair<int,int> cur=q.top(); q.pop(); if(--cur.second>0&&cur.first>time) q.push(cur);//数量大于0且第二天还能吃,就push回去 res++; } time++; } return res; } };
前面是pair
后面是vector
方法二:贪心+MapTree
map的底层通过红黑树实现,是一颗平衡的二叉搜索树,因此键是有序的(从小到大),因此可以使用map代替优先队列进行自动排序。
public: int eatenApples(vector<int>& apples, vector<int>& days) { int res=0; int time=0; int n=apples.size(); map<int,int> map; while(time<n||!map.empty()){ if(time<n&&apples[time]>0) map[time+days[time]-1]+=apples[time]; while(!map.empty()&&time>map.begin()->first) map.erase(map.begin()->first); if(!map.empty()){ if(--map.begin()->second==0) map.erase(map.begin()->first); res++; } time++; } return res; } };