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

题目描述

思路

当我们找到一个没有出现过的数的时候，将之前某个重复出现的数增加成这个没有出现过的数。注意，这里 「之前某个重复出现的数」 是可以任意选择的，它并不会影响最终的答案，因为将 P 增加到 X 并且将 Q 增加到 Y，与将 P 增加到 Y 并且将 Q 增加到 X 都需要进行 (X + Y) - (P + Q) 次操作。

例如当数组 A 为 [1, 1, 1, 1, 3, 5] 时，我们发现有 3 个重复的 1，且没有出现过 2，4 和 6，因此一共需要进行 (2 + 4 + 6) - (1 + 1 + 1) = 9 次操作。

算法

首先统计出每个数出现的次数，然后从小到大遍历每个数 x：

如果 x 出现了两次以上，就将额外出现的数记录下来（例如保存到一个列表中）；

如果 x 没有出现过，那么在记录下来的数中选取一个 v，将它增加到 x，需要进行的操作次数为 x - v。

我们还可以对该算法进行优化，使得我们不需要将额外出现的数记录下来。还是以 [1, 1, 1, 1, 3, 5] 为例，当我们发现有 3 个重复的 1 时，我们先将操作次数减去 1 + 1 + 1。接下来，当我们发现 2，4 和 6 都没有出现过时，我们依次将操作次数增加 2，4 和 6。
注意事项

虽然 A[i] 的范围为 [0, 40000)，但我们有可能会将数据递增到 40000 的两倍 80000。这是因为在最坏情况下，数组 A 中有 40000 个 40000，这样要使得数组值唯一，需要将其递增为 [40000, 40001, …, 79999]，因此用来统计的数组需要开到 80000。

代码

class Solution {
public:
    int minIncrementForUnique(vector<int>& A) {
        int cnt[80000] = {0};
        for(int i = 0; i < A.size(); i++)
        {
            cnt[A[i]]++;
        }
        int move = 0,taken = 0;
        for(int i = 0; i < 80000; i++)
        {
            if(cnt[i] >= 2)
            {
                taken += cnt[i] - 1;
                move -= i*(cnt[i] - 1);
            }
            else if(cnt[i] == 0 && taken > 0)
            {
                taken--;
                move += i;
            }
        }
        return move;
    }
};