思路:
贪心算法,每次挑选出满足启动资金(capital)要求的各个项目中利润(profit)最大的那个。
数据结构方面,考虑到有相同的capital值存在的可能,所以用multimap。且multimap如果按照key值删除,那么会一次性删除所有的相同key,不符合要求,所以采用迭代器。而且map的key值默认按照升序排列,可以减少运算量。
另外每次available的profit存于优先队列中,优先队列大的值先出。
AC代码:
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;
}
};