LeetCode287. 寻找重复数（二分）

1、题目描述

https://leetcode-cn.com/problems/find-the-duplicate-number/

给定一个包含 n + 1 个整数的数组 nums ，其数字都在 1 到 n 之间（包括 1 和 n），可知至少存在一个重复的整数。

假设 nums 只有 一个重复的整数 ，找出 这个重复的数 。

• nums.length == n + 1
• 1 <= nums[i] <= n
• nums 中 只有一个整数 出现 两次或多次 ，其余整数均只出现 一次

进阶：

• 如何证明 nums 中至少存在一个重复的数字?
• 你可以在不修改数组 nums 的情况下解决这个问题吗？
• 你可以只用常量级 O(1) 的额外空间解决这个问题吗？
• 你可以设计一个时间复杂度小于 O(n2) 的解决方案吗？

输入：nums = [1,3,4,2,2]
输出：2

输入：nums = [3,1,3,4,2]
输出：3

2、代码详解

使用二分法查找一个有范围的整数（结合抽屉原理）

使用的前提是，题目中说：

• 给定一个包含 n + 1 个整数的数组 nums，其数字都在 1 到 n 之间（包括 1 和 n）。
• 如果测试数据不在这个范围里，二分法失效。

class Solution(object):
    def findDuplicate(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        n = len(nums)
        left = 1
        right = n - 1

        while left < right:
            mid = left + (right - left)//2
            count = 0
            for num in nums:
                if num <= mid:
                    count += 1
            if count > mid:
                right = mid  # 重复的元素一定出现在 [left, mid] 区间里
            else:
                left = mid + 1  # [mid + 1, right]
        return left

s = Solution()
nums = [1, 3, 4, 2, 2]
print(s.findDuplicate(nums))

https://leetcode-cn.com/problems/find-the-duplicate-number/solution/er-fen-fa-si-lu-ji-dai-ma-python-by-liweiwei1419/