算法思路借鉴于官方题解,添加了详细注释用于理解,刷题打卡
写题解的过程用到了优先队列,随即想到其实现方法就是堆排序,翻书重新学习了一遍堆排序的算法,感觉以后二叉树系列的题可能会碰到0.0
typedef pair<int, int> dat;
class Solution
{
public:
int findMaximizedCapital(int k, int w, vector<int> &profits, vector<int> &capital)
{
vector<dat> proj;
for (int i = 0; i < profits.size(); i++)
{
proj.push_back({capital[i], profits[i]});//将利润和资本组合到一起
}
sort(proj.begin(), proj.end());//按照所需资本排序
int curr = 0;//记录当前放入堆中项目数
priority_queue<int, vector<int>, less<int>> q;//C++优先队列(最大堆)
for (int i = 0; i < k; i++)//退出循环分两种情况,K大于项目数或者选好K个项目
{
while (w >= proj[curr].first && curr < profits.size())//找出当前资本可选择的所有项目,放入最大堆排序
{
q.push(proj[curr].second);
curr++;
}
if (!q.empty())
{
w += q.top();//更新当前资本
q.pop();
}
else
break;//k大于项目数,所有能选的都被选掉,退出循环
}
return w;
}
};