给你一个整数数组 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(lowint l=low,h=high,p=a[low];while(lowwhile(low=p){high--;}a[low]=a[high];while(lowlow++;}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(lowint l=low,h=high,p=a[low];while(lowwhile(low=p){high--;}a[low]=a[high];while(lowlow++;}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;iif(nums[i]!=pre){nums[size++]=nums[i];pre=nums[i];}}// for(int i=0;i// printf("%d ",nums[i]);// }int sumcount=0;long long start=1;for(int i=0;iint 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(sumcountsum=sum+(start+start+k-sumcount-1)*(k-sumcount)/2;}return sum;}
下一篇:I/O复用--浅谈epoll