LeetCode 1705. 吃苹果的最大数目(优先队列)

 1705. 吃苹果的最大数目

有一棵特殊的苹果树,一连 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;
    }
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值