LeetCode 945. 使数组唯一的最小增量（C++、python）

最新推荐文章于 2022-04-23 20:38:18 发布

我很忙2010

最新推荐文章于 2022-04-23 20:38:18 发布

阅读量500

点赞数

分类专栏： LeetCode

本文链接：https://blog.csdn.net/qq_27060423/article/details/87922245

版权

LeetCode 专栏收录该内容

713 篇文章 10 订阅

订阅专栏

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

C++

class Solution {
public:
    int minIncrementForUnique(vector<int>& A) 
    {
        int n=A.size();
        if(0==n)
        {
            return 0;
        }
        sort(A.begin(),A.end());
        int res=0;
        for(int i=1;i<n;i++)
        {
            if(A[i]<=A[i-1])
            {
                int dif=A[i-1]-A[i]+1;
                A[i]=A[i-1]+1;
                res+=dif;                
            }
        }
        return res;       
    }
};

python

class Solution:
    def minIncrementForUnique(self, A: 'List[int]') -> 'int':
        n=len(A)
        if 0==n:
            return 0
        A.sort()
        res=0
        for i in range(1,n):
            if A[i-1]>=A[i]:
                dif=A[i-1]-A[i]+1
                A[i]=A[i-1]+1
                res+=dif
        return res