题目
《剑指offer》第三题
给定一个包含 n + 1 个整数的数组 nums,其数字都在 1 到 n 之间(包括 1 和 n),可知至少存在一个重复的整数。假设只有一个重复的整数,找出这个重复的数。
示例 1:
输入: [1,3,4,2,2]
输出: 2
示例 2:
输入: [3,1,3,4,2]
输出: 3
解法1 排序
将数组进行任意排序,便可以将重复元素安排在一起。然后通过一次遍历,当判定相邻元素相同的时候,便可以返回正确结果。
class Solution {
public:
int findDuplicate(vector<int>& nums) {
int len = nums.size();
if(len < 2)
return len - 1;
sort(nums.begin(),nums.end()); //C++ SORT 默认快排
for(int i = 1;i < len;i++)
{
if(nums[i] == nums[i - 1])
return nums[i];
}
return -1;
}
};
修改了数组本身。
时间复杂度
O
(
n
l
o
g
n
)
+
O
(
n
)
→
O
(
n
)
O(n logn) +O(n) \to O(n)
O(nlogn)+O(n)→O(n)
空间复杂度
O
(
1
)
O(1)
O(1) 快速排序的额外空间复杂度为O(1)
解法2 Hash set 查重
在遍历的同时构建hash set(key: num[i],value:i) ,每遍历到一个元素,首先在已有的hash set中进行查找,若查找到了,则当前元素即为重复元素。若未查找到,则将其插入,继续遍历。
class Solution {
public:
int findDuplicate(vector<int>& nums) {
unordered_map <int,int> num_set;
int len = nums.size();
if(len < 2)
return len - 1;
for(int i = 0 ; i < len ;i++)
{
auto hm = num_set.find(nums[i]);
if(hm == num_set.end())
num_set.insert(make_pair(nums[i],i));
else return hm->first;
}
return -1;
}
};
时间复杂度
O
(
n
)
O(n)
O(n) 一次遍历
空间复杂度 $O(n) $ 额外空间为hash set
未更改原数组。
解法3 快慢指针
我们可以假设数组中存在m个不同的值。将这m个不同的值看作链表的m个不同节点(0作为额外前驱节点)。数组中任意元素可以看做i到num[i]的关系。当出现重复元素时候,链表也就有了环。这也就转换成了链表是否有环的问题。
设置快慢指针,起点为链表的第一个节点,即为nums[0]. 快指针一次走两步,慢指针一次走一步。数组中存在重复元素,则链表必然有环。则快慢指针最终会相遇。
设从链表头到环衔接处长度为N,环的长度为K,设相遇在距离起点
(
N
+
x
)
(N + x)
(N+x)处。则有:
2
∗
(
N
∗
x
)
=
N
+
K
+
x
2* (N * x) = N + K + x
2∗(N∗x)=N+K+x
N
+
x
=
K
N + x = K
N+x=K
可以推出当相遇时,快指针距离下一次到达环还有N的距离,与链表头到环衔接出距离相等。所以我们可以设一个新的final指针指向链表的正式头节点与快指针继续向下走。当两者相遇时,相遇处就是正式的重复地点。
class Solution {
public:
int findDuplicate(vector<int>& nums) {
int len = nums.size();
int slow = nums[0],fast = nums[0];
do{
slow = nums[slow];
fast = nums[nums[fast]];
} while(slow != fast);
int start = nums[0];
while(start != fast)
{
fast = nums[fast];
start = nums[start];
}
return fast;
}
};
Q & A
当数组长度等于或者小于元素可去数值个数的时候,方法3是否还适用。
不适用!
如果数组长度为n,元素可取值也为1-n时候。我们需要0作为前驱节点,这样我们只有n-1个数组元素个数来表示n个数值,这是做不到的。