leetcode [287] 寻找重复数
Contact me:
Blog -> https://cugtyt.github.io/blog/index
Email -> cugtyt@qq.com
GitHub -> Cugtyt@GitHub
给定一个包含 n + 1 个整数的数组 nums,其数字都在 1 到 n 之间(包括 1 和 n),可知至少存在一个重复的整数。假设只有一个重复的整数,找出这个重复的数。
示例 1:
输入: [1,3,4,2,2]
输出: 2
示例 2:
输入: [3,1,3,4,2]
输出: 3
说明:
不能更改原数组(假设数组是只读的)。
只能使用额外的 O(1) 的空间。
时间复杂度小于 O(n2) 。
数组中只有一个重复的数字,但它可能不止重复出现一次。
来自题解:
由于数字范围在1-n,因此可以统计小于等于mid的数字个数,如果大于mid,表示重复数字肯定出现在这个区间,进一步缩小区间。
class Solution:
def findDuplicate(self, nums: List[int]) -> int:
low, high = 1, len(nums)
while low < high:
mid = (low + high) // 2
count = 0
for n in nums:
if n <= mid:
count += 1
if count > mid:
high = mid
else:
low = mid + 1
return low