一、题目描述
假设 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;
}
};