leetcode 945 Minimum Increment to Make Array Unique

1.题目描述

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

2.解题思路

进行排序后，利用数组递增（或递减）关系，实现仅判断当前数字与其前一位数字的比较。

3.Python代码

class Solution:
    def minIncrementForUnique(self, A):
        """
        :type A: List[int]
        :rtype: int
        """
        if A==[]:
            return 0
        s=sum(A)
        A=sorted(A)
        res，i=0， 1
        max_num=A[0]
        while i<len(A):
            if A[i]<=max_num:
                res+=A[i]-max_num+1
                A[i]=max_num+1
                max_num=max_num+1
            else:
                max_num=A[i]
            i+=1
        return(sum(A)-s)