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: 优先伫列
  • 相关阅读:
    easy ui 表单ajax和from两种提交数据方法
    easy ui 下拉级联效果 ,下拉框绑定数据select控件
    easy ui 下拉框绑定数据select控件
    easy ui 异步上传文件,跨域
    easy ui 菜单和按钮(Menu and Button)
    HTTP 错误 404.3
    EXTJS4.2 后台管理菜单栏
    HTML 背景图片自适应
    easy ui 表单元素input控件后面加说明(红色)
    EXTJS 4.2 添加滚动条
  • 原文地址:https://www.cnblogs.com/lightwindy/p/9859934.html
Copyright © 2011-2022 走看看