算法练习第十题——寻找重复数(不修改数组)

129 阅读2分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第11天,点击查看活动详情

算法练习第十题——寻找重复数(不修改数组)

寻找重复数(不修改数组)题目

给定一个包含 n + 1 个整数的数组 nums ,其数字都在 [1, n] 范围内(包括 1 和 n),可知至少存在一个重复的整数。

注意: 你设计的解决方案必须 不修改 数组 nums

示例一:

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

示例二:

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

示例三:

输入: nums = [0,1,5,4,0]
输出: 0

解题思路

方法一:二分查找

我们定义 cnt[i] 表示 nums 数组中小于等于 i 的数有多少个,假设我们重复的数是 target,那么 [1,target−1]里的所有数满足 cnt[i]≤i,[target,n] 里的所有数满足 cnt[i]>i,具有单调性。

以下代码以golang为例子:

func findDuplicate(nums []int) int {
    n := len(nums)
    l, r := 1, n - 1
    ans := -1
    for l <= r {
        mid := (l + r) >> 1
        cnt := 0
        for i := 0; i < n; i++ {
            if nums[i] <= mid {
                cnt++
            }
        }
        if cnt <= mid {
            l = mid + 1
        } else {
            r = mid - 1
            ans = mid
        }
    }
    return ans
}

该解法复杂度分析

  • 时间复杂度:O(nlogn),其中 n 为 nums 数组的长度。

  • 空间复杂度:O(1)。

方法三:快慢指针

我们对 nums 数组建图,每个位置 i 连一条 i→nums[i] 的边。由于存在的重复的数字 target,因此 target 这个位置一定有起码两条指向它的边,因此整张图一定存在环,且我们要找到的 target 就是这个环的入口,那么整个问题就等价于环形链表。

func findDuplicate(nums []int) int {
    slow, fast := 0, 0
    for slow, fast = nums[slow], nums[nums[fast]]; slow != fast; slow, fast = nums[slow], nums[nums[fast]] { }
    slow = 0
    for slow != fast {
        slow = nums[slow]
        fast = nums[fast]
    }
    return slow
}

该解法复杂度分析

  • 时间复杂度:O(n)O(n)。「Floyd 判圈算法」时间复杂度为线性的时间复杂度。

  • 空间复杂度:O(1)O(1)。我们只需要常数空间存放若干变量。

(点击进入专栏查看详细教程)