Leetcode287. 寻找重复数——python

题目

给定一个包含 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) 。
数组中只有一个重复的数字，但它可能不止重复出现一次。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/find-the-duplicate-number

代码

1、暴力求解（最差时间复杂度等于O(n^2))

class Solution(object):
    def findDuplicate(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        res = 0
        n = len(nums) - 1
        for i in nums:
            cnt = 0
            for j in nums:
                if j == i:
                    cnt += 1
            if cnt > 1:
                return i

2、二分法查找（时间复杂度O(nlogn))

class Solution(object):
    def findDuplicate(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        res = 0
        n = len(nums) - 1
        l,r = 1,n+1
        while l < r:
            mid = int(l + (r - l)/2)
            cnt = 0
            for i in nums:
                if i <= mid:
                    cnt += 1
            if cnt > mid:
                r = mid
            elif cnt <= mid:
                l = mid + 1
        return r