题目链接:287. 寻找重复数
题目
给定一个包含 n + 1 个整数的数组 nums,其数字都在 1 到 n 之间(包括 1 和 n),可知至少存在一个重复的整数。假设只有一个重复的整数,找出这个重复的数。
说明:
不能更改原数组(假设数组是只读的)。
只能使用额外的 O(1) 的空间。
时间复杂度小于 O(n2) 。
数组中只有一个重复的数字,但它可能不止重复出现一次。
解法
如果做一个nums[i]->nums[nums[i]]
的映射,则数组成为一个链表,例如{1, 3, 4, 5, 6, 4, 2}
就构成了一个1->3->5->4->6->2->4->6->...
的链表,如图:
可以发现由于数组中存在重复元素所以链表形成了部分闭环,6为闭环入口。这时如果用两个指针在链表上前进,一个每次前进1步,另一个每次前进2步,不难预见两者必在闭环的某个位置相遇(这个位置不一定是入口处),此时快指针比慢指针多走数个闭环长度的距离。
设两者同时出发到相遇用时t
,从起点到入口的距离为d
,环的长度为c
,则快指针走了2t
的距离,慢指针走了t
的距离,又由快指针多走了一圈2t-t=kc
,得t=kc
,此时两者与入口相距(kc-d)%c
。此时用一个新指针再从起点出发,每次前进1步,经过时间d
到达入口处,此时慢指针与入口相距(kc-d+d)%c=0
,即慢指针也到达了入口处,新指针与慢指针在入口处相遇。也即相遇处的值即为重复值,问题得解。
时间复杂度O(n)
,空间复杂度O(1)
。
特殊情况讨论
- 下标为0的值:因为所有的数均在1-n之间,所以不会有指向起点的指针,但不影响形成闭环。(如果题目改变之后受到影响可以先全体减一最后返回时加一)
- 如果某处
num[i]=i
,则会形成自环,但同样不影响:如果i
为重复元素,则上述解法可以解决,只是闭环中只有一个元素;如果i
不为重复元素,则这个闭环不被包含在链表中,不影响求解。
代码
Java:
class Solution {
public int findDuplicate(int[] nums) {
int fast = 0;
int slow = 0;
while (true) {
fast = nums[nums[fast]];
slow = nums[slow];
if (fast == slow) {
fast = 0;
while (fast != slow) {
fast = nums[fast];
slow = nums[slow];
}
return slow;
}
}
}
}