这道题其实和第一题是很相似的,也是一种贪心的算法(话说这周怎么都是贪心和暴力)
简单来说就是对于happiness的数组进行排序,用一个变量记录就行了。
思路:对于数组从大到小排序,然后对于happiness[i]-i进行取值累加。
注意:当我们遇到负数的时候不要加,直接放弃就行,因为题目中已经明确说明了,需要注意。循环的终止条件就是当i 上代码: 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; else sum+=happiness[i]-i; } return sum; }};
上代码:
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; else sum+=happiness[i]-i; } return sum; }};
京公网安备 11010502049817号