这道题其实和第一题是很相似的,也是一种贪心的算法(话说这周怎么都是贪心和暴力)
简单来说就是对于happiness的数组进行排序,用一个变量记录就行了。
思路:对于数组从大到小排序,然后对于happiness[i]-i进行取值累加。
注意:当我们遇到负数的时候不要加,直接放弃就行,因为题目中已经明确说明了,需要注意。循环的终止条件就是当i<k的时候结束就行了。
上代码:
class Solution {
public:long long maximumHappinessSum(vector<int>& happiness, int k) {int n=happiness.size();sort(happiness.begin(),happiness.end(),greater<int>());long long sum=0;;for(int i=0;i<k;i++){if(happiness[i]-i<0)break;elsesum+=happiness[i]-i;}return sum;}
};