leetcode_945 使数组唯一的最小增量

82 阅读1分钟

要求

给你一个整数数组 nums 。每次 move 操作将会选择任意一个满足 0 <= i < nums.length 的下标 i,并将 nums[i] 递增 1。

返回使 nums 中的每个值都变成唯一的所需要的最少操作次数。

示例 1:

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

示例 2:

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

提示:

  • 1 <= nums.length <= 105
  • 0 <= nums[i] <= 105

核心代码

class Solution:
    def minIncrementForUnique(self, nums: List[int]) -> int:
        nums.sort()
        ans = 0
        for i in range(1,len(nums)):
            if nums[i - 1] >= nums[i]:
                move = nums[i - 1] - nums[i] + 1
                nums[i] += move
                ans += move
        return ans

image.png

解题思路:我们先对nums进行排序,排好序后,需要注意一下我们使用nums[i - 1] >= nums[i],之所以需要使用大于等于是因为,[1,1,1],如果我们只是使用=的话,我们[1,2,1],在下一次变换就变换不了了,所以我们使用大于等于[1,2,1], 2 - 1 + 1 = 2,我们的第三位的数字就变成了[1,2,3],以此类推,注意这个题目的思路。