leetcode945. 使数组唯一的最小增量(水题_没啥说的 )

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
通过次数17,131提交次数36,635

class Solution {
public:
    int minIncrementForUnique(vector<int>& A) {

        int ans = 0;
        int r = 0; 
        int len = A.size();
        if(len == 0) return ans;
        int Min,Max;
        Min = A[0];
        Max = A[0];
        for(int i=0;i<len;i++){
            if(A[i]>Max)
                Max = A[i];
            if(A[i]<Min)
                Min = A[i];
        }
        vector<int> bucket(Max*2+2,0);
        for(int i=0;i<len;i++){
            bucket[A[i]]++;
        }
        r = Min;
        for(int i=Min;i<=Max;i++){
            //cout<<" i"<<i<<endl;
            while(bucket[i]>1){
                if(r<i) r = i; 
                while(bucket[r]!=0){
                    // cout<<"r "<<r<<bucket[r]<<endl;
                    r++;
                }
                // cout<<r<<endl;
                ans += r - i;
                r++;
                bucket[i]--;
            }
        }
        return ans;
    }
};
发布了124 篇原创文章 · 获赞 47 · 访问量 7万+

猜你喜欢

转载自blog.csdn.net/ludan_xia/article/details/105039528