leetcode 3075

最新推荐文章于 2024-05-08 23:57:02 发布

Nahida_nora

最新推荐文章于 2024-05-08 23:57:02 发布

阅读量202

点赞数 2

分类专栏： C++ Sort 文章标签： leetcode 算法

本文链接：https://blog.csdn.net/weixin_43876597/article/details/136900911

版权

C++ 同时被 2 个专栏收录

43 篇文章 0 订阅

订阅专栏

Sort

9 篇文章 0 订阅

订阅专栏

leetcode 3075

题目

在这里插入图片描述

例子

在这里插入图片描述

思路

孩子的幸福值最低也是0，所以选择最大的值，被选孩子的幸福值最高。需要使用排序算法

代码实现

class Solution {
public:
    long long maximumHappinessSum(vector<int>& happiness, int k) {

        //升序
        sort(happiness.begin(), happiness.end(), greater<int>());
        
        long long ans = 0;
        
        for(int i=0, j=0; i<k; i++){
            if(j < happiness.size()){
                if(happiness[j]-i <= 0){
                    return ans;
                }else{
                    ans += happiness[j] -i;
                    j++;
                }
            }
        }
        return ans;      
    }
};

执行时间是177ms

class Solution {
public:
    long long maximumHappinessSum(vector<int>& happiness, int k) {

        //升序
        sort(happiness.begin(), happiness.end(), greater<int>());
        
        long long ans = 0;
        
        for(int i=0; i<k && happiness[i] > i; i++){
            ans += happiness[i] -i;
        }
        return ans;      
    }
};