945. 使数组唯一的最小增量（中等题）

题目描述：
给定整数数组 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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/minimum-increment-to-make-array-unique
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
解法：

class Solution {
    public int minIncrementForUnique(int[] A) {
        if (A == null || A.length <= 1) {
            return 0;
        }
        int[] count = new int[80000];
        for (int i : A) {
            count[i]++;
        }
        int res = 0;
        int temp = 0;
        for (int i = 0; i < count.length; i++) {
            if (count[i] >= 2) {
                temp += count[i] - 1;
                res -= i * (count[i] - 1);
            } else if (count[i] == 0 && temp > 0) {
                temp--;
                res += i;
            }
        }
        return res;
    }
}

思路概述：
先用count[ ]数组记录每个数出现了几次，之所以把数组长度定位80000，是因为A[ ]数组长度最大为40000，元素小于40000，如果出现40000个39999，操作后最大是79999，所以长度定为80000，防止溢出。
之后遍历count[ ] 数组，每出现个数大于2的情况，我们就用temp记录下有几个操作的元素，即 count[ i ] - 1,比如当前重复数字是2，下一个空位置是4，那么我们需要加1的操作次数就是4-2。所以我们定义res记录加一的次数，先让res减去需要操作的个数个的当前元素，也就是res -= i * (count[i] - 1);之后只要temp>0，即还有重复元素，和count[i]=0，即当前元素个数为0，就让res加上i，之后让temp减一即可。