给你一个长度为 n
的数组 happiness
,以及一个 正整数 k
。
n
个孩子站成一队,其中第 i
个孩子的 幸福值 是 happiness[i]
。你计划组织 k
轮筛选从这 n
个孩子中选出 k
个孩子。
在每一轮选择一个孩子时,所有 尚未 被选中的孩子的 幸福值 将减少 1
。注意,幸福值 不能 变成负数,且只有在它是正数的情况下才会减少。
选择 k
个孩子,并使你选中的孩子幸福值之和最大,返回你能够得到的 最大值 。
示例 1:
输入:happiness = [1,2,3], k = 2 输出:4 解释:按以下方式选择 2 个孩子: - 选择幸福值为 3 的孩子。剩余孩子的幸福值变为 [0,1] 。 - 选择幸福值为 1 的孩子。剩余孩子的幸福值变为 [0] 。注意幸福值不能小于 0 。 所选孩子的幸福值之和为 3 + 1 = 4 。
示例 2:
输入:happiness = [1,1,1,1], k = 2 输出:1 解释:按以下方式选择 2 个孩子: - 选择幸福值为 1 的任意一个孩子。剩余孩子的幸福值变为 [0,0,0] 。 - 选择幸福值为 0 的孩子。剩余孩子的幸福值变为 [0,0] 。 所选孩子的幸福值之和为 1 + 0 = 1 。
示例 3:
输入:happiness = [2,3,4,5], k = 1 输出:5 解释:按以下方式选择 1 个孩子: - 选择幸福值为 5 的孩子。剩余孩子的幸福值变为 [1,2,3] 。 所选孩子的幸福值之和为 5 。
提示:
1 <= n == happiness.length <= 2 * 10^5
1 <= happiness[i] <= 10^8
1 <= k <= n
我的解答:
class Solution {public long maximumHappinessSum(int[] happiness, int k) {// 对数组排序Arrays.sort(happiness);int size = happiness.length - 1;long res = 0;// 每次都选择幸福值最高的孩子for(int i = 0; i < k; i++){// 每次选择孩子的剩余幸福值等于自身幸福值 - 选择次数,但不小于0res += Math.max(0,happiness[size - i] - i);}return res;}
}