【冲冲冲】贪心来当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 刷题笔记分类整理不同算法类型,也有不同语言版本,对于菜鸡的我来说帮助很大。

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

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

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 211,042评论 6 490
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 89,996评论 2 384
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 156,674评论 0 345
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,340评论 1 283
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,404评论 5 384
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,749评论 1 289
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,902评论 3 405
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,662评论 0 266
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,110评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,451评论 2 325
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,577评论 1 340
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,258评论 4 328
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,848评论 3 312
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,726评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,952评论 1 264
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,271评论 2 360
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,452评论 2 348

推荐阅读更多精彩内容

  • 汇总几个常见的贪心算法实现的问题 概述 IPO(最大投资收益) 金砖最小分割代价 会议室相关问题 分发糖果 柠檬水...
    yaco阅读 1,415评论 0 2
  • 502. IPO 给定若干个项目。对于每个项目 i,它都有一个纯利润 Pi,并且需要最小的资本 Ci 来启动相应的...
    李海游阅读 140评论 0 0
  • 502 IPO IPO Description:Suppose LeetCode will start its I...
    air_melt阅读 259评论 0 1
  • 萌矿主起家之前传。 导语:资本永不眠,金钱是流动的饿鬼,饥渴难耐,永不满足的攫取利润。只有穷过,才知金钱短缺的饥渴...
    萌矿主阅读 584评论 0 0
  • 货币 亚当· 斯密论分工。分工能增进生产力的原因:1.使人们的技艺日进;2.节约从一个工种转换到另一个工种的时间;...
    开过头先生阅读 1,435评论 0 0