3075. 幸福值最大化的选择方案
解法
思路和算法
为了使选择的 $k$ 个孩子的幸福值之和最大,应遵循如下贪心策略:按照幸福值递减的顺序选择幸福值最大的 $k$ 个孩子。理由如下。
-
如果将幸福值最大的 $k$ 个孩子中的任意一个孩子更换成一个幸福值较小的孩子,则更换之后的孩子的幸福值一定不变或减少,不可能增加,因此幸福值之和不可能更大。
-
当选择幸福值最大的 $k$ 个孩子时,如果这 $k$ 个孩子的初始幸福值都不小于 $k - 1$ 则幸福值之和的总减少量等于 $\dfrac{k(k + 1)}{2}$,如果这 $k$ 个孩子中存在孩子的初始幸福值小于 $k - 1$ 则幸福值之和的总减少量小于 $\dfrac{k(k + 1)}{2}$。假设两个孩子的幸福值分别为 $x$ 和 $y$,其中 $x > y$,则先选择 $x$ 后选择 $y$ 的幸福值之和等于 $x + \max(y - 1, 0)$,先选择 $y$ 后选择 $x$ 的幸福值之和等于 $y + \max(x - 1, 0)$,由于 $x > y > 0$,因此 $\max(x - 1, 0) = x - 1$,$x + \max(y - 1, 0) \ge y + \max(x - 1, 0)$,即先选择 $x$ 后选择 $y$ 的方案更优。
根据贪心策略,计算选中的孩子的幸福值之和的最大值的方法如下。
-
将数组 $\textit{happiness}$ 按升序排序。
-
用 $n$ 表示数组 $\textit{happiness}$ 的长度,反向遍历排序后的数组 $\textit{happiness}$,对于每个 $1 \le i \le k$,需要选择幸福值为 $\textit{happiness}[n - i]$ 的孩子,该孩子被选择时的幸福值为 $\max(\textit{happiness}[n - i] - (i - 1), 0)$,计算选中的孩子的幸福值之和。
实现方面,反向遍历排序后的数组 $\textit{happiness}$ 时,当遇到 $\textit{happiness}[n - i] \le i - 1$ 时,其余孩子的幸福值一定都是 $0$,此时即可结束遍历。
遍历结束之后,即可得到选中的孩子的幸福值之和的最大值。
代码
###Java
class Solution {
public long maximumHappinessSum(int[] happiness, int k) {
long sum = 0;
Arrays.sort(happiness);
int n = happiness.length;
for (int i = 1; i <= k && happiness[n - i] > i - 1; i++) {
int curr = happiness[n - i] - (i - 1);
sum += curr;
}
return sum;
}
}
###C#
public class Solution {
public long MaximumHappinessSum(int[] happiness, int k) {
long sum = 0;
Array.Sort(happiness);
int n = happiness.Length;
for (int i = 1; i <= k && happiness[n - i] > i - 1; i++) {
int curr = happiness[n - i] - (i - 1);
sum += curr;
}
return sum;
}
}
复杂度分析
-
时间复杂度:$O(n \log n)$,其中 $n$ 是数组 $\textit{happiness}$ 的长度。将数组 $\textit{happiness}$ 排序的时间是 $O(n \log n)$,排序后遍历数组 $\textit{happiness}$ 计算选中的孩子的幸福值之和的最大值的时间是 $O(n)$,因此时间复杂度是 $O(n \log n)$。
-
空间复杂度:$O(\log n)$,其中 $n$ 是数组 $\textit{happiness}$ 的长度。排序的递归调用栈空间是 $O(\log n)$。