【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;} };總結
以上是生活随笔為你收集整理的【LeetCode 502】IPO的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 【LeetCode】732. 我的日程安
- 下一篇: 【LeetCode】414.第三大的数