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

https://leetcode-cn.com/problems/minimum-increment-to-make-array-unique/

题目

给定整数数组 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解法

1. 暴力法

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

2. 排序法

class Solution {
public:
    int minIncrementForUnique(vector<int>& A) {
        sort(A.begin(),A.end());
        int ret=0,token=0;
        for(int i=1;i<A.size();i++)
            if(A[i]==A[i-1]){
                ret-=A[i];
                token++;
            }else{
                int give=min(token,A[i]-A[i-1]-1);
                ret+=give*(give+1)/2+give*A[i-1];
                token-=give;
            }
        if(token>0)
            ret+=token*A[A.size()-1]+token*(token+1)/2;
        return ret;        
    }
};