给你一个整数数组 nums 和一个整数 k 。请你向 nums 中追加 k 个 未 出现在 nums 中的、互不相同 的 正 整数,并使结果数组的元素和 最小 。
返回追加到 nums 中的 k 个整数之和。
示例 1:
输入:nums = [1,4,25,10,25], k = 2
输出:5
解释:在该解法中,向数组中追加的两个互不相同且未出现的正整数是 2 和 3 。
nums 最终元素和为 1 + 4 + 25 + 10 + 25 + 2 + 3 = 70 ,这是所有情况中的最小值。
所以追加到数组中的两个整数之和是 2 + 3 = 5 ,所以返回 5 。
示例2:
输入:nums = [5,6], k = 6
输出:25
解释:在该解法中,向数组中追加的两个互不相同且未出现的正整数是 1 、2 、3 、4 、7 和 8 。
nums 最终元素和为 5 + 6 + 1 + 2 + 3 + 4 + 7 + 8 = 36 ,这是所有情况中的最小值。
所以追加到数组中的两个整数之和是 1 + 2 + 3 + 4 + 7 + 8 = 25 ,所以返回 25 。
提示:
- 1 <= nums.length <= 10^5
- 1 <= nums[i], k <= 10^9
参考代码:
//分割为多段,等差数列求和
class Solution {
public long minimalKSum(int[] nums, int k) {
// TreeSet:排序、消除重复元素
Set<Integer> set = new TreeSet<>();
for (int num : nums) {
set.add(num);
}
long ans = 0;
// 从1开始尝试填加
int start = 1;
for (int num : set) {
int cnt = Math.min(k, num - start);
ans += (long)(start + start + cnt - 1) * cnt / 2;
k -= cnt;
start += cnt;
// 跳过Set中的元素
if (start == num) {
++start;
}
}
// nums范围内已使用的数字个数不足k,继续添加最小的一段连续数字
if (k > 0) {
ans += (long)(start + start + k - 1) * k / 2;
}
return ans;
}
}