题目
给定整数数组 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的数值进行规则操作,直到数值个数等于原数组元素个数,累计每次操作总次数。
- 将原数组分类为数值和对应个数的字典
- 循环检查每个数值的个数,大于1则向前推进一个数值
- 直到所有数值个数等于原数组中元素个数
- 时间复杂度:O(n2)
- 空间复杂度:O(n)
当前算法对于大部分测试用例通过了,在下面的测试用例超时了,该测试用例有40000(4万)个整数:
https://www.zhenxiangsimple.com/files/tech/testCase20200322.txt
class Solution:
def minIncrementForUnique(self, A: List[int]) -> int:
cnt,lenA = 0,len(A)
dictA = collections.Counter(A)
while