题目描述:
给定一个未排序的整数数组,找出最长连续序列的长度。
要求算法的时间复杂度为 O(n)。
示例:
输入: [100, 4, 200, 1, 3, 2]
输出: 4
解释: 最长连续序列是 [1, 2, 3, 4]。它的长度为 4。
AC C++ Solution:
方法1:
利用现成的STL标准库算法,先进行sort()排序,然后unique()删除重复元素。最后判断相邻元素是否连续。
不过这种方案虽然简单,但时间复杂度比较高,执行用时为20ms。
代码:
class Solution {
public:
int longestConsecutive(vector<int>& nums) {
int n = nums.size();
if(n < 2)
return n;
sort(nums.begin(),nums.end());
unique(nums.begin(),nums.end());
int cnt = 1;
int maxl = 1;
for(int i = 1; i < n; i++) {
if(nums[i] - nums[i-1] == 1) {
cnt++;
maxl = max(cnt,maxl);
}
else
cnt = 1;
}
return maxl;
}
};
方法2:
使用unordered_set容器(此处不使用set容器,因为set是默认有序的,初始化时耗时多),
对于数组中的每个元素,在set中查找相邻的数字是否存在。
class Solution {
public:
int longestConsecutive(vector<int> &num) {
if(num.empty())
return 0;
unordered_set<int> record(num.begin(),num.end());
int res = 1;
for(int n : num){
if(record.find(n)==record.end()) continue;
record.erase(n);
int prev = n-1,next = n+1;
while(record.find(prev)!=record.end()) record.erase(prev--); //查找一次后就删除,避免重复查找
while(record.find(next)!=record.end()) record.erase(next++);
res = max(res,next-prev-1);
}
return res;
}
};