有一棵特殊的苹果树,一连 n 天,每天都可以长出若干个苹果。在第 i 天,树上会长出 apples[i] 个苹果,这些苹果将会在 days[i] 天后(也就是说,第 i + days[i] 天时)腐烂,变得无法食用。也可能有那么几天,树上不会长出新的苹果,此时用 apples[i] == 0 且 days[i] == 0 表示。
你打算每天 最多 吃一个苹果来保证营养均衡。注意,你可以在这 n 天之后继续吃苹果。
给你两个长度为 n 的整数数组 days 和 apples ,返回你可以吃掉的苹果的最大数目。
先贴一个自己写的代码,思路还是有的,就是将所有苹果数量和坏掉的日期记录下来,看起来代码没问题,也可以保证先吃先坏的苹果,但是呢,WA了114514次
class Solution {
public int eatenApples(int[] apples, int[] days) {
int ans=0,apple=0;
//<最后可以吃的日子:明天就坏掉的苹果数量>
HashMap<Integer,Integer>map=new HashMap<>();
//因为遍历map不能修改map值,所以改为遍历keyset
List<Integer>keys=new ArrayList<>();
int n=apples.length;
for(int i=0;i<n;i++){
int num=map.getOrDefault(i+days[i],0);
map.put(i+days[i],num+apples[i]);
keys.add(i+days[i]);
}
Collections.sort(keys);
for(int key:keys){
System.out.println("剩下的苹果最多放到第"+key+"天"+
"还剩下"+map.getOrDefault(key,0)+"个");
}
while(!map.isEmpty()){
for(Integer key:keys){
if(key>ans&&map.containsKey(key)){
if(map.get(key)>0){
System.out.println("吃了一个苹果,保质期到第"+key+"天");
map.put(key, map.get(key)-1);
ans++;
}else{
map.remove(key);
}
break;
}else{
map.remove(key);
}
}
}
return ans;
}
}
错误的原因在于,吃苹果和处理数据并不是两个完全独立的过程,反而有着很大的关系,如果把两个过程割裂开来看,很可能就会出现奇怪的情况,如果第一天长得果子保质期久第二天的果子保质期短,可能第一天就不吃而是等待第二天再吃了,这就很不合理,因此我们要把维护数据结构和吃苹果放在一个循环里,同时处理
class Solution {
public int eatenApples(int[] apples, int[] days) {
int ans = 0;
Queue<int[]> queue = new PriorityQueue<>((x, y) -> x[0]-y[0]);
int d = 0;
while (d < apples.length || !queue.isEmpty()){
if (d < apples.length && apples[d] > 0) {
queue.add(new int[]{d+days[d], apples[d]});// 新结出的苹果入队
}
while (!queue.isEmpty() && (queue.peek()[0] <= d || queue.peek()[1] == 0)) {
queue.poll();
} // 到腐烂日期或者腐烂数量归零
if (!queue.isEmpty()) { // 吃苹果
++ans;
--queue.peek()[1];
}
++d;
}
return ans;
}
}