Description
给定整数数组 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/minimum-increment-to-make-array-unique
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
Solution 1
贪心。
class Solution:
def minIncrementForUnique(self, A: List[int]) -> int:
# 贪心算法
A.sort()
count = 0
for i in range(1, len(A)):
if A[i] <= A[i - 1]: # 当前元素比上一个元素小要增加本元素
count += A[i - 1] - A[i] + 1 # 需要增加的次数
A[i] = A[i - 1] + 1 # 当前元素要比上一个+1
return count
Solution 2
用空间换时间。参考方法二思想
class Solution:
def minIncrementForUnique(self, A: List[int]) -> int:
hashmap = [0 for _ in range(50000)]
cnt = 0
for n in A:
hashmap[n] += 1
for i in range(len(hashmap)):
if hashmap[i] > 1:
hashmap[i+1] += hashmap[i] - 1
cnt += hashmap[i] - 1
return cnt