【LeetCode 502】IPO

一、题目描述

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

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

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

二、示例

示例 1:

输入: 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 位有符号整数范围内。

三、分析

贪心算法：每次挑选出满足启动资金(capital)要求的各个项目中利润(profit)最大的那个。

数据结构方面：因为存在相同的capital值的可能，所以用multimap。但是multimap如果按照key值删除，那么将会删除所有相同key的数据，不满足题意，所以采用迭代器。
而且map的key值默认按照升序排列，可以减少运算量。

此外priority_queue自带排序功能

四、实现

class Solution {  
public:  
    int findMaximizedCapital(int k, int W, vector<int>& Profits, vector<int>& Capitals) {
        
        multimap<int,int> map;
        for(int i = 0; i < Capitals.size(); ++i){
            map.insert(make_pair(Capitals[i],Profits[i]));
        }
        
        priority_queue<int> prique;
        while(k--){
            for(auto iter = map.begin(); iter != map.end(); iter++){//map的key自动按照升序排列
                if(iter->first <= W){
                    prique.push(iter->second);//优先队列里面存的是available的利润值
                    map.erase(iter);
                }else break;
            }
            if(prique.empty()) break;
            W += prique.top();//优先队列top出来的是最大的数.
            prique.pop();//此轮在队列中留下的数在下一轮继续available
        }
        
        return W;
    }  
};