erlang算法系列-leetclode-2195. 向数组中追加 K 个整数（中等）

最新推荐文章于 2024-07-23 22:06:37 发布

不秃头的IT者

最新推荐文章于 2024-07-23 22:06:37 发布

阅读量134

点赞数

分类专栏： leetcode解题 erlang算法文章标签： erlang 算法 leetcode

本文链接：https://blog.csdn.net/smallbire/article/details/123339428

版权

leetcode解题同时被 2 个专栏收录

36 篇文章 0 订阅

订阅专栏

erlang算法

36 篇文章 0 订阅

订阅专栏

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

解题思路

1、先从小到大进行排序；
2、计算每两个之间整数的和sum(Ni ~ N(i+1)),确保只计算K个整数。

-spec minimal_k_sum(Nums :: [integer()], K :: integer()) -> integer().
minimal_k_sum(Nums, K) ->
    L = lists:sort(Nums),
    do(L, 1, K, 0).
  
do(L, I, 0, Ans) ->
    Ans;
do([], I, K, Ans) ->
    Ans + (I+K) * (I+K-1) div 2 - (I)*(I-1) div 2;
do([N | L], I, K, Ans) ->
    if
        I > N ->
            do(L, I, K, Ans);
        N =:= I ->
            do(L, I+1, K, Ans);
        true ->
            {Ans1, K1 } =do2(I,N-1, K),
           do(L, N+1, K1, Ans+Ans1)
    end.
do1(K) ->
    K*(k+1) div 2.

do2(I, N, K) ->
    N1 = min(K, N-I+1),
    Ans = (I+N1)*(I+N1-1) div 2 - I*(I-1) div 2,
    {Ans, K - N1}.