今天看啥  ›  专栏  ›  大梦三千秋

LeetCode 使数组唯一的最小增量

大梦三千秋  · 简书  ·  · 2020-03-22 21:34

使数组唯一的最小增量


题目来源: 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

解题思路


思路:贪心算法

具体操作:先遍历,然后再比较计数

根据题意,通过自增 1,消除重复值。因为要先对数组进行排序,当最小的元素是唯一的情况下,则没必要自增。

题意中,提及返回使 A 中的每个值都是唯一的最少操作次数。所以自增是有上限的,当增加到某个值就不必要继续增加。

具体以下图解:

图解

代码实现


class Solution:
    def minIncrementForUnique(self, A: List[int]) -> int:
        A_len = len(A)
        if A_len == 0:
            return 0
        
        A.sort()

        res = 0
        cur_num = A[0]

        for i in range(1, A_len):
            if A[i] >= cur_num + 1:
                cur_num = A[i]
            else:
                res += (cur_num + 1 - A[i])
                cur_num += 1
        
        return res

实现结果


实现结果

以上就是使用贪心算法,先遍历再计数来解决 《使数组唯一的最小增量》问题的主要内容。


欢迎关注微信公众号《书所集录》




原文地址:访问原文地址
快照地址: 访问文章快照