美文网首页leetcode每日打卡
【冲冲冲】贪心来当leetcode CEO

【冲冲冲】贪心来当leetcode CEO

作者: TripleEyeAline | 来源:发表于2021-09-08 11:12 被阅读0次

    假设 力扣(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 刷题笔记分类整理不同算法类型,也有不同语言版本,对于菜鸡的我来说帮助很大。

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

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

    相关文章

      网友评论

        本文标题:【冲冲冲】贪心来当leetcode CEO

        本文链接:https://www.haomeiwen.com/subject/kifvwltx.html