Skip to the content.

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