【2195. 向数组中追加K个整数】排序+贪心算法

 

方法一：排序+贪心+枚举到最大可能的边界，没有使用高斯求和，没有去重，是通过while循环判断的，会超时！！

//会出现数组中有两个19的情况！！即数组中出现重复元素的情况
//下面方法会超时！
class Solution {
public:
    long long minimalKSum(vector<int>& nums, int k) {
        sort(nums.begin(),nums.end());
        int len=nums.size();
        // int max_one=max(nums[len-1],k);这个是错的！示例2就会出错！
        int max_one=len+k;
        int point=0,num_add=0;
        long long sum=0;
        for(int i=1;i<=max_one;i++)
        {
            if(num_add==k)
                break;
            if(point==len||nums[point]!=i)//注意先后顺序！否则point=len时会溢出
            {
                num_add++;
                sum+=i;
            }
            else
            {
                while(point!=len&&nums[point]==i)
                {
                    point++;
                }
            }
        }
        return sum;
    }
};

 方法二：排序+去重+贪心+高斯求和

class Solution {
public:
    long long minimalKSum(vector<int>& nums, int k) {
        sort(nums.begin(),nums.end());
        int len=unique(nums.begin(),nums.end())-nums.begin();
        long long sum=0;
        for(int i=0;i<len;i++)
        {
            if(nums[i]<=k)//注意等于号！！
            {
                sum+=nums[i];
                k++;
            }
        }
        long long ans=k*(k+1LL)/2-sum;//要把1写成1LL，否则按int算，乘完会溢出！！
        return ans;
    }
};

 方法三：排序+逐步判断重复+逐步添加+贪心+高斯求和

class Solution {
public:
    long long minimalKSum(vector<int>& nums, int k) {
        nums.push_back(0);
        nums.push_back(int(2e9) + 1);
        sort(nums.begin(), nums.end());

        long long presum = 0, ans = 0;
        for (int i = 1; i < nums.size(); ++i) {
            int offer = nums[i] - nums[i - 1] - 1;
            if (offer > 0) {
                if (offer < k) {
                    k -= offer;
                }
                else {
                    ans = static_cast<long long>(nums[i - 1] + k + 1) * (nums[i - 1] + k) / 2 - presum;
                    break;
                }
            }
            if (nums[i] != nums[i - 1]) {
                presum += nums[i];
            }
        }

        return ans;
    }
};

作者：LeetCode-Solution
链接：https://leetcode-cn.com/problems/append-k-integers-with-minimal-sum/solution/xiang-shu-zu-zhong-zhui-jia-k-ge-zheng-s-9vdv/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。