2195. 向数组中追加 K 个整数-快速排序暴力解法+优化解法

2195. 向数组中追加 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 。

暴力解法解题代码如下：

void quick(int *a,int low,int high){
    if(low<high){
        int l=low,h=high,p=a[low];
        while(low<high){
            while(low<high&&a[high]>=p){
                high--;
            }
            a[low]=a[high];
           
            while(low<high&&a[low]<=p){
                low++;
            }
            a[high]=a[low];
           
        }
        a[low]=p;
       
        quick(a,l,low-1);
        quick(a,low+1,h);
    }
}

long long minimalKSum(int* nums, int numsSize, int k){
    quick(nums,0,numsSize-1);
    int now=1;
    long long sum=0;
    
    int i=0;
   
    while(k>0){
       
        if(nums[i]==now){
            int start=nums[i];
            while(start==nums[i]){
                            i++;
                   if(i==numsSize){
                                break;
                            }
            }
        
           
        }else{
            
            k--;
            sum=sum+now;
        }
         now++;
        if(i==numsSize){
            break;
        }
        
    }
    while(k>0){
        k--;
         printf("%d ",sum);
        sum=sum+now;
         now++;
    }
  
    return sum;

}

上面算法无法通过时间，这一题其实我觉得难度是很大的，使用c语言去编程的话，优化后的代码如下：

void quick(int *a,int low,int high){
    if(low<high){
        int l=low,h=high,p=a[low];
        while(low<high){
            while(low<high&&a[high]>=p){
                high--;
            }
            a[low]=a[high];
           
            while(low<high&&a[low]<=p){
                low++;
            }
            a[high]=a[low];
           
        }
        a[low]=p;
       
        quick(a,l,low-1);
        quick(a,low+1,h);
    }
}
int cmp(const void* a, const void* b)
 {
     return *(int*)a - *(int*)b;
 }

long long minimalKSum(int* nums, int numsSize, int k){
 
       qsort(nums, numsSize, sizeof(int), cmp);
   
    long long sum=0;
    int size=0;
    int pre=nums[0]+1;
    for(int i=0;i<numsSize;i++){
        if(nums[i]!=pre){
            nums[size++]=nums[i];
            pre=nums[i];
        }
       
    }
    //   for(int i=0;i<size;i++){
    //       printf("%d ",nums[i]);
    //   }
      int sumcount=0;
      long long start=1;
       for(int i=0;i<size;i++){
         
            int count=nums[i]-start;
           int presumcount=sumcount;
          sumcount=count+sumcount;
          
          if(sumcount>=k){
           //   printf("zz %d %d %d %d %d %d ",sum,presumcount,sumcount,start,k,nums[i]);
                sum=sum+(start+start+k-presumcount-1)*(k-presumcount)/2;
              break;
          

          }
          else{
              sum=sum+(start+nums[i]-1)*count/2;

          }
          start=nums[i]+1;
       
            
      }
    


        
      if(sumcount<k){
            sum=sum+(start+start+k-sumcount-1)*(k-sumcount)/2;

      }
    return sum;

}