1.题目
给定一个包含 n + 1 个整数的数组 nums,其数字都在 1 到 n 之间(包括 1 和 n),可知至少存在一个重复的整数。假设只有一个重复的整数,找出这个重复的数。
示例 1:
输入: [1,3,4,2,2]
输出: 2
示例 2:
输入: [3,1,3,4,2]
输出: 3
说明:
1.不能更改原数组(假设数组是只读的)。
2.只能使用额外的 O(1) 的空间。
3.时间复杂度小于 O(n2) 。
4.数组中只有一个重复的数字,但它可能不止重复出现一次。
2.思路
一维数组,寻找某个数,可以想到用二分查找。
(1) 首先确定边界,由题得知,其数字都在 1 到 n 之间,所以,left = 1, right = n
(2) 计算mid = left +(right- left) / 2 .
(3) 确定遍历条件,因为只有一个重复的数,当left=right时,即可跳出循环。 所以遍历条件 left < right
(4)循环遍历统计小于等于mid的元素个数, 假设 mid = 4,那么不大于mid的个数应该 <= 4, 若大于mid,则说明该区间内存在重复元素,缩小区间范围,right = mid , 重复上述统计过程。
反之,若小于mid,则说明另一区间存在重复元素,缩小区间范围,left = mid + 1
class Solution:
def findDuplicate(self, nums: List[int]) -> int:
n = len(nums)
left = 1
right = n
while left < right:
mid = left + right >> 1
cnt = 0
for num in nums:
if num <= mid:
cnt += 1
if cnt > mid:
right = mid
else:
left = mid + 1
return left