leetcode+IPO 小安晋升分享（第502题哦）

最新推荐文章于 2019-09-03 10:39:01 发布

原创最新推荐文章于 2019-09-03 10:39:01 发布 · 223 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode 专栏收录该内容

29 篇文章

订阅专栏

本文探讨了LeetCode IPO前的项目选择策略，旨在通过完成最多k个项目以最大化资本。采用贪婪算法，结合两个优先队列，一个按利润排序，另一个按启动资本排序，实现了资本的最大化。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

IPO

今天是小安开始Leetcode刷题的第502题，正文开始ing?

题目描述

假设 LeetCode 即将开始其 IPO。为了以更高的价格将股票卖给风险投资公司，LeetCode希望在 IPO 之前开展一些项目以增加其资本。由于资源有限，它只能在 IPO 之前完成最多 k 个不同的项目。帮助 LeetCode 设计完成最多 k 个不同项目后得到最大总资本的方式。

给定若干个项目。对于每个项目 i，它都有一个纯利润 Pi，并且需要最小的资本 Ci 来启动相应的项目。最初，你有 W 资本。当你完成一个项目时，你将获得纯利润，且利润将被添加到你的总资本中。

总而言之，从给定项目中选择最多 k 个不同项目的列表，以最大化最终资本，并输出最终可获得的最多资本。题目原址
示例

输入: k=2, W=0, Profits=[1,2,3], Capital=[0,1,1].
输出: 4
解释:
由于你的初始资本为 0，你尽可以从 0 号项目开始。
在完成后，你将获得 1 的利润，你的总资本将变为 1。
此时你可以选择开始 1 号或 2 号项目。
由于你最多可以选择两个项目，所以你需要完成 2 号项目以获得最大的资本。
因此，输出最后最大化的资本，为 0 + 1 + 3 = 4。

注意

假设所有输入数字都是非负整数。
表示利润和资本的数组的长度不超过 50000。
答案保证在 32 位有符号整数范围内。

方法：贪婪算法

思路

创建两个优先队列，一个按照收益从大到小排，一个按启动资金从小到大排。

选择使用一个优先队列 profitQueue，按照利润 Pi 从高到低，将我们当前 W 资本能够启动的项目添加进入
选择使用一个优先队列 capitalQueue，按照启动资本 Ci 从低到高，将剩余的项目添加进入
profitQueue 取出利润最大者剔除，然后更新当前资本 W，资本 W 更新了就需要将当前资本 W 能够启动的项目从 capitalQueue 中移到 profitQueue 中
重复上一步，执行 k 次后最终获得最优解

代码实现

【Java实现】

class Solution {
    public int findMaximizedCapital(int k, int W, int[] Profits, int[] Capital) {
        // 这个队列按照收益从大到小排序，且只包含能启动的项目
        PriorityQueue<Project> profitQueue = new PriorityQueue<>((o1, o2) -> o2.profits - o1.profits);
        // 这个队列按照启动资本从小到大排序，且只包含目前资本 W 不能启动的项目
        PriorityQueue<Project> capitalQueue = new PriorityQueue<>((o1, o2) -> o1.capital - o2.capital);
        for (int i = 0; i < Profits.length; i++) {
            if (Capital[i] > W) {
                capitalQueue.add(new Project(Profits[i], Capital[i]));
            } else {
                profitQueue.add(new Project(Profits[i], Capital[i]));
            }
        }
        for (int i = 0; i < k; i++) {
            if (profitQueue.isEmpty()) {
                break; // profitQueue 为空，即没有能够启动的项目了，结束循环
            }
            W += profitQueue.poll().profits; // 取出收益最大的元素，并更新资本 W
            while (!capitalQueue.isEmpty() && W >= capitalQueue.peek().capital) {
                // 将 capitalQueue 能够启动的项目加入到 profitQueue 中
                profitQueue.offer(capitalQueue.poll());
            }
        }
        return W;
    }
    
    static class Project {
        int profits;
        int capital;
        public Project(int profits, int capital) {
            this.profits = profits;
            this.capital = capital;
        }
    }
}