6017.向数组中追加K个整数(java)

力扣题 6017.向数组中追加K个整数

给你一个整数数组 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 。

来源：力扣（LeetCode）
链接：6017.向数组中追加K个整数

class Solution {
    public long minimalKSum(int[] nums, int k) {  
        //sum记录最小元素和
        //head记录未出现区间的左值
        long sum = 0,head = 1;
        //给nums数组排序
        Arrays.sort(nums);
        for(int i=0;i<nums.length;i++){
            //判断是否需要追加
            if(head<nums[i]){
                //need记录未出现区间的长度，head+need-1即为未出现区间的右值
                //当nums[i]-head>=k,说明该范围大于等于需要追加的K，need只需要取k值即行
                int need = (int)(nums[i]-head) >= k ? k:(int)(nums[i]-head);
                //等差数列计算head与head+need-1之间的值
                sum += (2*head+need-1)*need/2;
                k=k-need;
            }
            head = nums[i]+1;
        }
        //当k不为0时，nums最大值之前的未 出现在 nums 中的、互不相同的正整数数已全部算入sum,
        //但是追加的数仍不够k，继续追加
        if(k>0){
            int need = k;
            sum += (2*head+need-1)*need/2;
        }
        return sum;
    }
}