LC-2195 Medium LeetCode

2195. Append K Integers With Minimal Sum

Read the full problem statement on LeetCode.

Difficulty: medium Acceptance: 26% Topics: Array, Math, Greedy, Sorting

Reading material

Arrays & Hashing 12 min Math & Geometry 12 min Greedy 11 min

Reference solution (spoiler · python)

# Time:  O(nlogn)
# Space: O(n)

# greedy
class Solution(object):
    def minimalKSum(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: int
        """
        result = k*(k+1)//2
        curr = k+1
        for x in sorted(set(nums)):
            if x < curr:
                result += curr-x
                curr += 1
        return result


# Time:  O(nlogn)
# Space: O(n)
# greedy
class Solution2(object):
    def minimalKSum(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: int
        """
        result = prev = 0
        nums.append(float("inf"))
        for x in sorted(set(nums)):
            if not k:
                break
            cnt = min((x-1)-prev, k)
            k -= cnt
            result += ((prev+1)+(prev+cnt))*cnt//2
            prev = x
        return result

Solution from kamyu104/LeetCode-Solutions · MIT