✨贪心+大根堆解决——>力扣502. IPO

299 阅读2分钟

小知识,大挑战!本文正在参与“程序员必备小知识”创作活动

Code皮皮虾 一个沙雕而又有趣的憨憨少年,和大多数小伙伴们一样喜欢听歌、游戏,当然除此之外还有写作的兴趣,emm...,日子还很长,让我们一起加油努力叭🌈



🌝题目

力扣——>原题链接

在这里插入图片描述



✨解题思路

因为题目说了,最多选K个,输出最终可获得的最多资本,那么我们需要在有限的次数获取更多的资本,🔥即让每一次获取都尽可能地多

那么如何让每一次获取都尽可能地多呢?

答案就是:维护一个大根堆,每次获取都去获取顶部的元素也就是最大的资本,当然还有一个前提:就是获取该项目的资本要小于等于当前的我们的资本



🔥代码

class Solution {
    public int findMaximizedCapital(int k, int w, int[] profits, int[] capital) {
        int n = profits.length;
        

        //对启动资本和纯利润建立二维数组
        int[][] nums = new int[n][2];

        for (int i = 0; i < n; ++i) {
            nums[i][0] = capital[i];
            nums[i][1] = profits[i];
        }

        //利用排序,根据nums[0] 也就是启动资本 进行升序排序
        Arrays.sort(nums, (a, b) -> a[0] - b[0]);


        //建立大根堆
        PriorityQueue<Integer> queue = new PriorityQueue<>((x, y) -> y - x);

        int index = 0;

        //k次获取
        for (int i = 0; i < k; ++i) {
            
            //如果当前的启动资本小于等于则代表是我们可以获取的,那么把他对应的利润加入到堆中去
            while (index < n && nums[index][0] <= w) {
                queue.add(nums[index][1]);
                index += 1;
            }
            //当堆不为空才去poll元素
            if (!queue.isEmpty()) {
                //poll返回堆顶元素也就是可获取的最大利润,加入到w中
                w += queue.poll();
            } else {
                break;
            }
        }

        return w;
    }
}


🔥精选专栏

毛遂自荐,给大家推荐一下自己的专栏😁,欢迎小伙伴们收藏关注😊

力扣算法题解专区

小白学Java

MybatisPlus专栏

App爬虫专栏

PC端爬虫专栏

大厂面试题专栏


💖最后

我是 Code皮皮虾,一个热爱分享知识的 皮皮虾爱好者,未来的日子里会不断更新出对大家有益的博文,期待大家的关注!!!

创作不易,如果这篇博文对各位有帮助,希望各位小伙伴可以==一键三连哦!==,感谢支持,我们下次再见~~~


一键三连.png