zoukankan      html  css  js  c++  java
  • [LeetCode] 502. IPO 上市

    Suppose LeetCode will start its IPO soon. In order to sell a good price of its shares to Venture Capital, LeetCode would like to work on some projects to increase its capital before the IPO. Since it has limited resources, it can only finish at most k distinct projects before the IPO. Help LeetCode design the best way to maximize its total capital after finishing at most k distinct projects.

    You are given several projects. For each project i, it has a pure profit Pi and a minimum capital of Ci is needed to start the corresponding project. Initially, you have W capital. When you finish a project, you will obtain its pure profit and the profit will be added to your total capital.

    To sum up, pick a list of at most k distinct projects from given projects to maximize your final capital, and output your final maximized capital.

    Example 1:

    Input: k=2, W=0, Profits=[1,2,3], Capital=[0,1,1].
    
    Output: 4
    
    Explanation: Since your initial capital is 0, you can only start the project indexed 0.
                 After finishing it you will obtain profit 1 and your capital becomes 1.
                 With capital 1, you can either start the project indexed 1 or the project indexed 2.
                 Since you can choose at most 2 projects, you need to finish the project indexed 2 to get the maximum capital.
                 Therefore, output the final maximized capital, which is 0 + 1 + 3 = 4. 

    Note:

    1. You may assume all numbers in the input are non-negative integers.
    2. The length of Profits array and Capital array will not exceed 50,000.
    3. The answer is guaranteed to fit in a 32-bit signed integer.

    假设LeetCode即将开始其首次公开募股。为了向Venture Capital出售其股票的优惠价格,LeetCode希望在IPO之前开展一些项目以增加其资本。由于资源有限,它只能在IPO之前完成最多k个不同的项目。帮助LeetCode设计在完成最多k个不同项目后使得资本最多的最佳方法。

    你有几个项目。 对于每个项目i,它具有纯利润Pi,并且需要最小资本Ci来启动相应的项目。 最初你有W资本,完成项目后,您将获得纯利润,利润将被添加到您的总资本中。总之,从给定项目列表中选择最多k个不同项目,最大化最终资本,并输出您的最终最大化的资本。

    解法:Priority Queue,保持当前可能的最大利润。 一旦达到所需资金就插入可能的利润。

    Java:

    public class Solution {
        public int findMaximizedCapital(int k, int W, int[] Profits, int[] Capital) {
            PriorityQueue<int[]> pqCap = new PriorityQueue<>((a, b) -> (a[0] - b[0]));
            PriorityQueue<int[]> pqPro = new PriorityQueue<>((a, b) -> (b[1] - a[1]));
            
            for (int i = 0; i < Profits.length; i++) {
                pqCap.add(new int[] {Capital[i], Profits[i]});
            }
            
            for (int i = 0; i < k; i++) {
                while (!pqCap.isEmpty() && pqCap.peek()[0] <= W) {
                    pqPro.add(pqCap.poll());
                }
                
                if (pqPro.isEmpty()) break;
                
                W += pqPro.poll()[1];
            }
            
            return W;
        }
    }
    

    Python:

    def findMaximizedCapital(self, k, W, Profits, Capital):
            heap = []
            projects = sorted(zip(Profits, Capital), key=lambda l: l[1])
            i = 0
            for _ in range(k):
                while i < len(projects) and projects[i][1] <= W:
                    heapq.heappush(heap, -projects[i][0])
                    i += 1
                if heap: W -= heapq.heappop(heap)
            return W  

    Python:

    def findMaximizedCapital(self, k, W, Profits, Capital):
        current = []
        future = sorted(zip(Capital, Profits))[::-1]
        for _ in range(k):
            while future and future[-1][0] <= W:
                heapq.heappush(current, -future.pop()[1])
            if current:
                W -= heapq.heappop(current)
        return W  

    C++:

    class Solution {
    public:
        int findMaximizedCapital(int k, int W, vector<int>& Profits, vector<int>& Capital) {
            priority_queue<pair<int, int>> maxH;
            priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>>> minH;
            for (int i = 0; i < Capital.size(); ++i) {
                minH.push({Capital[i], Profits[i]});
            }
            for (int i = 0; i < k; ++i) {
                while (!minH.empty() && minH.top().first <= W) {
                    auto t = minH.top(); minH.pop();
                    maxH.push({t.second, t.first}); 
                }
                if (maxH.empty()) break;
                W += maxH.top().first; maxH.pop();
            }
            return W;
        }
    };
    

    C++:  

    class Solution {
    public:
        int findMaximizedCapital(int k, int W, vector<int>& Profits, vector<int>& Capital) {
            priority_queue<int> q;
            multiset<pair<int, int>> s;
            for (int i = 0; i < Capital.size(); ++i) {
                s.insert({Capital[i], Profits[i]});
            }
            for (int i = 0; i < k; ++i) {
                for (auto it = s.begin(); it != s.end(); ++it) {
                    if (it->first > W) break;
                    q.push(it->second);
                    s.erase(it);
                }
                if (q.empty()) break;
                W += q.top(); q.pop();
            }
            return W;
        }
    };
    

      

      

    All LeetCode Questions List 题目汇总

    queue  [kjuː]  详细X
    基本翻译
    n. 队列;长队;辫子
    vt. 将…梳成辫子;使…排队
    vi. 排队;排队等候
    网络释义
    Queue: 队列
    Message Queue: 消息队列
    priority queue: 优先伫列
  • 相关阅读:
    爱加密亮相第十八届软博会,移动App安全引关注
    android 实现自己定义状态栏通知(Status Notification)
    Android中View绘制流程以及invalidate()等相关方法分析
    设计模式 ( 十八 ) 策略模式Strategy(对象行为型)
    目标检測的图像特征提取之(一)HOG特征
    Vbox 未指定XXX网络名称 找不到网卡问题
    NetBeans工具学习之道:NetBeans IDE Java 高速新手教程
    NETSH WINSOCK RESET这条命令的含义和作用?
    红帽/CentOS ext4无法格式化大分区 补充ext4格式化方式
    android之PackageManager简单介绍
  • 原文地址:https://www.cnblogs.com/lightwindy/p/9859934.html
Copyright © 2011-2022 走看看