题目:Given an unsorted array of integers, find the length of the longest consecutive elements sequence.
Your algorithm should run in O(n) complexity.
Example:
Input: [100, 4, 200, 1, 3, 2]
Output: 4
Explanation: The longest consecutive elements sequence is [1, 2, 3, 4]
. Therefore its length is 4.
题目分析:
求最长连续串,直观想法是排序,但题目的难度在于要求O(n)时间复杂度,排序最快是O(nlgn),不满足要求。
方法1:考虑使用unordered_set(基于哈希表),存储数组中的元素。遍历数组,查找集合。若在集合中存在该数,则在集合中删除,并在集合中查找其前一个数和后一个数,若有,则继续删除,并继续向前向后查找,直到不存在,则继续遍历数组。
class Solution {
public:
int longestConsecutive(vector<int>& nums) {
unordered_set<int> nums_set(nums.begin(), nums.end());//存入数组到集合中;
int length=0;
for(int i=0;i<nums.size();i++){
if(!nums_set.count(nums[i])){//查找集合,若不存在,继续遍历;
continue;
}
nums_set.erase(nums[i]);//若存在,删除该数;
int pre=nums[i]-1;//查找其前一项和后一项;
int pro=nums[i]+1;
while(nums_set.count(pre)){
nums_set.erase(pre);
pre--;
}
while(nums_set.count(pro)){
nums_set.erase(pro);
pro++;
}
if(length<pro-pre-1){//记录当前最长串长度;
length=pro-pre-1;
}
}
return length;
}
};
方法2:考虑使用unordered_map,存储数组中的元素及其连续元素个数。遍历数组,修改该数的连续元素个数。若该数连续元素个数被设置过,则跳过;否则通过其前后元素的连续元素个数计算该数连续元素个数,并修改该连续元素的边界值(连续序列的中间已经不会再被使用,不必修改;例如6 3 5 4 ,在修改4时要修改3 6而不必修改5)
代码如下:
class Solution {
public:
int longestConsecutive(vector<int>& nums) {
unordered_map<int, int> m;
int length=0;
for(int i=0;i<nums.size();i++){
int t=nums[i];
if(m[t]){
continue;
}
m[t] =m[t + 1] + m[t - 1] + 1;
m[t + m[t + 1]] = m[t - m[t - 1]] =m[t + 1] + m[t - 1] + 1;
length=max(length,m[t]);
}
return length;
}
};