题目地址:
https://www.acwing.com/problem/content/description/15/
给定一个长度为 n + 1 n+1 n+1的数组 A A A,数组中所有的数均在 1 ∼ n 1∼n 1∼n的范围内,其中 n ≥ 1 n≥1 n≥1。请找出数组中任意一个重复的数,但不能修改输入的数组。
数据范围:
1
≤
n
≤
1000
1≤n≤1000
1≤n≤1000
考虑 0 → A [ 0 ] → A [ A [ 0 ] ] → . . . 0\to A[0]\to A[A[0]]\to ... 0→A[0]→A[A[0]]→...,设 0 , A [ 0 ] , . . . , A k [ 0 ] 0,A[0],...,A^k[0] 0,A[0],...,Ak[0]两两不同,但是 A i [ 0 ] = A k + 1 [ 0 ] , 0 < i ≤ k A^i[0]=A^{k+1}[0], 0<i\le k Ai[0]=Ak+1[0],0<i≤k,所以 A i − 1 [ 0 ] ≠ A k [ 0 ] , A [ A i − 1 [ 0 ] ] = A [ A k [ 0 ] ] A^{i-1}[0]\ne A^k[0],A[A^{i-1}[0]]=A[A^k[0]] Ai−1[0]=Ak[0],A[Ai−1[0]]=A[Ak[0]],这相当于含环单链表求环的入口问题,可以用Floyd算法,参考https://blog.csdn.net/qq_46105170/article/details/104015181。代码如下:
#include <vector>
using namespace std;
class Solution {
public:
int duplicateInArray(vector<int>& nums) {
int f = 0, s = 0;
do {
f = nums[nums[f]];
s = nums[s];
} while (f != s);
f = 0;
while (f != s) {
f = nums[f];
s = nums[s];
}
return s;
}
};
时间复杂度 O ( n ) O(n) O(n),空间 O ( 1 ) O(1) O(1)。