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

Posted 2021-09-24 Code皮皮虾

文章目录

• • 😉毛遂自荐
  • 🌝题目
  • ✨解题思路
  • 🔥代码
  • 💖最后

😉毛遂自荐

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

大厂面试题专栏

Java专栏

爬虫专栏

更多专栏尽在主页，点我😁！！！

---

🌝题目

力扣——>原题链接

---

✨解题思路

因为题目说了，最多选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;
    }
}

---

💖最后

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

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

---