mirror of
https://github.com/ShusenTang/LeetCode.git
synced 2024-09-02 14:20:01 +00:00
86 lines
4.3 KiB
Markdown
86 lines
4.3 KiB
Markdown
![]() |
# [128. Longest Consecutive Sequence](https://leetcode.com/problems/longest-consecutive-sequence/)
|
|||
|
|
|||
|
# 思路
|
|||
|
|
|||
|
给定一个无序的数组,找出最长的可组成连续序列的元素的个数,要求时间复杂度为O(n)。
|
|||
|
|
|||
|
## 思路一
|
|||
|
|
|||
|
看看暴力法怎样解此题:遍历一遍数组,对每个数num,我们假设num就是某个连续范围的最小值,所以我们需要判断num+1、num+2...是否在数组中。可见这样的时间复杂度为O(n^2)。
|
|||
|
|
|||
|
我们可以对暴力法优化一下:
|
|||
|
* 由于要经常判断某个元素是否在数组中,所以我们可以事先用一个hash存放每个元素,查找复杂度为O(1);
|
|||
|
* 对于某个元素num,如果 num-1 也在数组中,直接跳过就可以了,因为我们处理 num-1 时就会处理到num、num+1...。即如果num是某个候选范围的起点时(即num-1不在数组中),我们才不断判断num+1、num+2...是否在数组中。
|
|||
|
|
|||
|
虽然有两层循环,但是每个范围只会被遍历一次,所以总的复杂度为O(n)
|
|||
|
|
|||
|
## 思路二
|
|||
|
|
|||
|
这题其实可以看做是区间合并的问题,可以采取类似并查集的思路。我们用一个名为mp的hashmap记录以这个数为端点(可能左也可能右端点)的连续范围的长度,例如果`mp[4] = 2`,说明存在一个以4为左端点的长度为2的连续范围(即4、5),或者存在一个以4为右端点的长度为2的连续范围(即3、4)。但是值得注意的是,如果数num不在某个范围的端点,那么`mp[num]`的值没有意义,我们也不会用它。
|
|||
|
|
|||
|
那么如果我们遍历到了数num,该怎样求得`mp[num]`呢。
|
|||
|
* 首先如果之前已经遇到过num了,那直接跳过即可;
|
|||
|
* 否则我们可以查看num-1和num+1是否之前出现过:
|
|||
|
* 如果出现过num-1,那么`mp[num-1]`一定表示以num-1为右端点的范围长度(因为num之前没出现过,所以num-1不可能是左端点,下同);
|
|||
|
* 如果出现过num+1,那么`mp[num+1]`一定表示以num+1为左端点的范围长度;
|
|||
|
|
|||
|
所以num可把左右两个范围连成一个大小为`mp[num-1] + mp[num+1] + 1`的范围,因此需要更新该大范围的左端点的值`mp[num - mp[num-1]]`和右端点的值`mp[num + mp[num+1]]`。至于`mp[num]`等于什么无关紧要,因为我们不需要用除了端点之外的值,所以我们随便赋一个值给`mp[num]`就行了,仅仅表明已经出现num了。
|
|||
|
|
|||
|
本思路相比于思路一的优势在于无需先将所有元素用一个hash存着,所以适合数组动态增长的情况。
|
|||
|
|
|||
|
只需要遍历一遍数组,所以总的时间复杂度为O(n)
|
|||
|
|
|||
|
# C++
|
|||
|
## 思路一
|
|||
|
``` C++
|
|||
|
class Solution {
|
|||
|
public:
|
|||
|
int longestConsecutive(vector<int>& nums) {
|
|||
|
unordered_set<int>num_set(nums.begin(), nums.end());
|
|||
|
int n = num_set.size();
|
|||
|
if(n <= 1) return n;
|
|||
|
|
|||
|
int res = 1;
|
|||
|
for(int num: num_set){
|
|||
|
if(!num_set.count(num - 1)){ // 如果num是起点
|
|||
|
int cur_res = 1;
|
|||
|
while(num_set.count(num + cur_res)) cur_res++;
|
|||
|
res = max(cur_res, res);
|
|||
|
}
|
|||
|
}
|
|||
|
return res;
|
|||
|
}
|
|||
|
};
|
|||
|
```
|
|||
|
|
|||
|
## 思路二
|
|||
|
``` C++
|
|||
|
class Solution {
|
|||
|
public:
|
|||
|
int longestConsecutive(vector<int>& nums) {
|
|||
|
//mp[num] = i 表示以num为端点(可能左也可能右端点)的连续范围的长度
|
|||
|
unordered_map<int, int>mp;
|
|||
|
int n = nums.size();
|
|||
|
if(n <= 1) return n;
|
|||
|
|
|||
|
int res = 1;
|
|||
|
for(int num: nums){
|
|||
|
if(mp.count(num)) continue; // 这表明num已经出现过
|
|||
|
|
|||
|
// 如果 mp.count(num-1) == 1 说明以num-1为右端点的连续范围长度为mp[num-1]
|
|||
|
int l = mp.count(num - 1) ? mp[num - 1] : 0;
|
|||
|
// 如果 mp.count(num+1) == 1 说明以num+1为左端点的连续范围长度为mp[num-1]
|
|||
|
int r = mp.count(num + 1) ? mp[num + 1] : 0;
|
|||
|
|
|||
|
int total = r + l + 1; // 左右两个连续范围组成一个大的连续范围
|
|||
|
res = max(res, total);
|
|||
|
|
|||
|
mp[num] = -1; // 这里可以赋成任意值, 因为我们不会用除了端点之外的值
|
|||
|
|
|||
|
mp[num - l] = total; // 更新大范围的左端点的值
|
|||
|
mp[num + r] = total; // 更新大范围的右端点的值
|
|||
|
}
|
|||
|
return res;
|
|||
|
}
|
|||
|
};
|
|||
|
```
|