持续创作,加速成长!这是我参与「掘金日新计划 · 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)。我们只需要常数空间存放若干变量。