Question

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

English Version

题目描述

给你一个整数数组 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] <= nums[i - 1]:
        d = nums[i - 1] - nums[i] + 1
        nums[i] += d
        ans += d
    return ans

class Solution {
  public int minIncrementForUnique(int[] nums) {
    Arrays.sort(nums);
    int ans = 0;
    for (int i = 1; i < nums.length; ++i) {
      if (nums[i] <= nums[i - 1]) {
        int d = nums[i - 1] - nums[i] + 1;
        nums[i] += d;
        ans += d;
      }
    }
    return ans;
  }
}

class Solution {
public:
  int minIncrementForUnique(vector<int>& nums) {
    sort(nums.begin(), nums.end());
    int ans = 0;
    for (int i = 1; i < nums.size(); ++i) {
      if (nums[i] <= nums[i - 1]) {
        int d = nums[i - 1] - nums[i] + 1;
        nums[i] += d;
        ans += d;
      }
    }
    return ans;
  }
};

func minIncrementForUnique(nums []int) int {
  sort.Ints(nums)
  ans := 0
  for i := 1; i < len(nums); i++ {
    if nums[i] <= nums[i-1] {
      d := nums[i-1] - nums[i] + 1
      nums[i] += d
      ans += d
    }
  }
  return ans
}