【数据结构与算法 - 练习题】数组：使数组唯一的最小增量

【数据结构与算法 - 练习题】数组：使数组唯一的最小增量

题目描述

给定整数数组 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

解题思路

1. 首先将数组进行排序，然后从左到右遍历数组。
2. 如果当前元素大于上一个元素，保持不变；如果当前元素小于等于上一个元素，就需要增加当前元素，直到大于上一个元素。
3. 时间复杂度：O(nlogn)

代码

public int minIncrementForUnique(int[] A) {
    Arrays.sort(A);

    int count = 0;
    for (int i = 1; i < A.length; i++) {
        if (A[i] <= A[i-1]) {
            count += A[i-1] - A[i] + 1;
            A[i] = A[i-1] + 1;
        }
    }
    return count;
}

代码注意点

临界值

1. 数组A = null：本题条件：0 <= A.length <= 40000，故A不可能为null, 可忽略。
2. A.length = 0 时代码是否正常。
3. 操作次数为int类型是否会存在溢出问题：题目中条件“0 <= A[i] < 40000”，操作次数最大为(1+4000) * 4000 / 2，不会溢出。

逻辑细节

1. 计数与数组元素更改顺序：一定要先根据A[i]和A[i-1]的值进行计数，再更改A[i]的值，否则A[i]的更改会使count计数错误。
2. 刚排序完成时，数组中前一个元素的值一定小于等于后一个元素的值，即A[i-1] <= A[i]。但是随着不符合条件的存在，数组中元素的值在不断的增长，可能会出现A[i-1] > A[i]的情况，故判断条件应该为：A[i] <= A[i-1]。

其他解题思路

由于题目中“0 <= A[i] < 40000”, 数组中元素的取值范围比较小，可以考虑使用计数的方法。

1. 例如输入 [3, 2, 1, 2, 1, 7]，计数之后有两个 1 和两个 2。我们先看最小的数，两个 1 重复了，需要有一个增加到2，这样 2 的数量变成了三个。在三个 2 中，又有两个需要增加到 3，然后又出现了两个 3…… 以此类推，可以计算出需要增加的次数。
2. 时间复杂度 O(n + k)。

题目来源

LeetCode: minimum-increment-to-make-array-unique (使数组唯一的最小增量)