有一棵特殊的苹果树,一连 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 才成立
题解:贪心加优先队列,往优先队列里存一个数组,第一个数为第几天,第二个数为几天之后坏,优先队列按照苹果腐烂的时间从小到大排序,首先遍历整个数组,每次取出优先队列的头部,比较当前的天数和腐烂的时间,一直找到优先队列为空或者能吃到苹果为止。 之后需要再对优先队列进行遍历,因为还可以继续吃苹果,从apples.length+1天开始。注意,数组下表是从0开始的,题目中苹果是从第一天开始算的,注意分别。
import java.util.Comparator;
import java.util.PriorityQueue;
class Solution {
public int eatenApples(int[] apples, int[] days) {
PriorityQueue<int[]> pq=new PriorityQueue<int[]>(new Comparator<int[]>() {
@Override
public int compare(int[] o1, int[] o2) {
return o1[0]+o1[1]-(o2[0]+o2[1]);
}
});
int ans=0;
for(int i=0;i<apples.length;i++)
{
pq.add(new int[] {i+1,days[i]});
while(!pq.isEmpty())
{
int t[]=pq.peek();
if(t[0]+t[1]>i+1&&apples[t[0]]>0)
{
ans++;
apples[t[0]]--;
break;
}
else
pq.poll();
}
}
int day=days.length+1;
while(!pq.isEmpty())
{
while(!pq.isEmpty())
{
int t[]=pq.peek();
if(t[0]+t[1]>day&&apples[t[0]]>0)
{
ans++;
apples[t[0]]--;
break;
}
else
{
pq.poll();
}
}
day++;
}
return ans;
}
}