使数组唯一的最小增量
题目来源: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
实现结果
以上就是使用贪心算法,先遍历再计数来解决 《使数组唯一的最小增量》问题的主要内容。
欢迎关注微信公众号《书所集录》
**粗体** _斜体_ [链接](http://example.com) `代码` - 列表 > 引用
。你还可以使用@
来通知其他用户。