【冲冲冲】贪心来当leetcode CEO

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

给你 n 个项目。对于每个项目 i ，它都有一个纯利润 profits[i] ，和启动该项目需要的最小资本 capital[i] 。

最初，你的资本为 w 。当你完成一个项目时，你将获得纯利润，且利润将被添加到你的总资本中。

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

答案保证在 32 位有符号整数范围内。

示例 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。

来源：力扣（LeetCode）

读题

1. 最多做 k 个不同项目
2. 做项目不花钱，只看自己的钱够不够成本。w >= capital[i]
   要求：最大化的最终持有资本。

读完题，就一个感觉，要干就干最能挣钱的，这不是让我贪心吗？好家伙，冲冲冲…

解题思路

讲个故事先

你是 LeetCode CEO，将要走向巅峰 IPO，为了想拉高股票价格多（割）赚（韭）钱（cai），你要在限定项目次数内尽可能的多挣钱。

你的初始资本为 w，做项目空手套白狼不花钱。需要注意的是你的资本 w 需要 >= 想干项目所需的成本，这项目只能干一次。

1. 首先，你只能干 k 个项目。那肯定得每次干的项目都得最赚钱，这样你的资本 w 就增长，持有资本越多就能干更大更赚钱的项目。
2. 好，咱来干第一票，得选所有能干的项目中最赚钱的，咋办？ 先把项目按照所需资本从小到大 sort 一下。然后做最赚钱的那个项目。咱做了最赚钱的项目，空手套白狼，不花钱。同时呢，咱初始资产 + 这第一票赚的钱，持有资本增长了，那不得赶紧物色下一个最赚钱的。
3. 之后操作就像这第一票一样，直到干完 k 次。
4. done，你本次临时当 CEO 所持有的的最终最大资本就出来了。

回到代码层面

1. 上边小故事里说了要把项目所需资金从小到大排序一下。
2. 用一个优先队列priority_queue<int> q，每次循环时候将项目所需资金小于所持资本 w 的项目利润放入优先队列 q，自动排序。每次取 q 的 top()，更新 w。然后如是这般循环 k 次即可。

贪心其实就是像这样每一个小阶段局部最优，然后就可以推到全局最优。就像这题每次干最赚钱的，最后就能得到最大持有资本。

PS:

• Q: C++ sort 对 vector<pair<int, int>> 优先比较 pair.first，然后比较 pair.second ?
• A: std::sort 可以不写比较函数，不写排序函数的话就用 operator< 来比较。而 std::pair::operator< 按标准，两个 std::pair 的 first 为第一优先级，second 为第二优先级。

template <class _Ty1, class _Ty2>
_NODISCARD constexpr bool operator<(const pair<_Ty1, _Ty2>& _Left, const pair<_Ty1, _Ty2>& _Right) {
    return _Left.first < _Right.first || (!(_Right.first < _Left.first) && _Left.second < _Right.second);
}

代码

class Solution {
public:
    int findMaximizedCapital(int k, int w, vector<int>& profits, vector<int>& capital) {
        vector<pair<int, int>> info;
        int size = profits.size();
        for(int i = 0; i < size; i++) {
            info.push_back({capital[i], profits[i]});
        }

        sort(info.begin(), info.end());
        priority_queue<int> q;
        int idx = 0, iSize = info.size();
        while(k--){
            while(idx < iSize && w >= info[idx].first) {
                q.push(info[idx++].second); 
            }
            if(q.empty()) {
                break; 
            }else {
                w += q.top(); 
                q.pop();
            }
        }
        return w;
    }
};

补充&鸣谢

在此非常感谢代码随想录。这位老哥的 leetcode 刷题笔记分类整理不同算法类型，也有不同语言版本，对于菜鸡的我来说帮助很大。

另外，我也整了个刷题笔记，坚持费曼学习法，吃进去，产出来。

如果感觉本题解对你有帮助，请不要吝啬点赞👍🏻啊 冲冲冲…