有一棵特殊的苹果树,一连 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 个苹果:
- 第一天到第三天,你吃的都是第一天长出来的苹果。
- 第四天和第五天不吃苹果。
- 第六天和第七天,你吃的都是第六天长出来的苹果。
提示:
apples.length == n
days.length == n
1 <= n <= 2 * 104
0 <= apples[i], days[i] <= 2 * 104
只有在 apples[i] = 0 时,days[i] = 0 才成立
题解 (贪心+优先队列)
贪心:优先吃最快要腐烂的苹果
小顶堆:根据苹果可以维持到的天数排序,顶部是剩余天数最小的苹果
特殊处理:
1.可持续到日期变为今天的的移除队列
2.根据苹果可以维持到的天数排序,不要根据苹果剩余的天数排序,根据剩余天数排序所有苹果每天剩余天数都要-1
3.优先队列存int数组而不是hashmap会更方便操作
4.注意第 n 天之后还可以继续吃苹果,所以要一直模拟到堆为空为止
class Solution {
public int eatenApples(int[] apples, int[] days) {
//a[0]是苹果可以持续到的日期,a[1]是苹果的数目
//小顶堆,a1[0]-a2[0]
PriorityQueue<int[]> queue=new PriorityQueue<>((a1,a2)->(a1[0]-a2[0]));
int n=apples.length;
int count=0;
//注意第 n 天之后还可以继续吃苹果,所以要一直模拟到堆为空为止
for(int i=0;i<n||!queue.isEmpty();i++){
//如果树上还可以产出苹果&&当天苹果个数>0,把当天的苹果放入队列
if(i<n&&apples[i]>0) queue.offer(new int[]{days[i]+i,apples[i]});
//取当天最快要腐烂的苹果
while(!queue.isEmpty()){
int[] tmp=queue.poll();
if(tmp[0]>i){
count++;
if(--tmp[1]>0) queue.offer(tmp);
break;
}
}
}
return count;
}
}