跳转至

LeetCode: 945. 使数组唯一的最小增量

1、题目描述

给定整数数组 A,每次 move 操作将会选择任意 A[i],并将其递增 1

返回使 A 中的每个值都是唯一的最少操作次数。

示例 1:

输入:[1,2,2]
输出:1
解释:经过一次 move 操作,数组将变为 [1, 2, 3]。

示例 2:

输入:[3,2,1,2,1,7]
输出:6
解释:经过 6 次 move 操作,数组将变为 [3, 4, 1, 2, 5, 7]。
可以看出 5 次或 5 次以下的 move 操作是不能让数组的每个值唯一的。

提示:

  • 0 <= A.length <= 40000
  • 0 <= A[i] < 40000

2、解题思路

增量最小,就表示找出比当前值大的数组中不存在的那个值即可

  • 先排序
  • 设置一个基准值,每次先更新基准值,然后更新当前值的增量
class Solution:
    def minIncrementForUnique(self, A: List[int]) -> int:
        A.sort()
        ans = 0
        base = 0
        for num in A:
            base = max(num, base)
            ans += base - num
            base += 1

        return ans