题目大意:求未排序数组中的最长连续序列的长度,要求时间复杂度为O(n)。例如[100,4,200,1,3,2],len=4。
分析:哈希表unordered_set。
原题=>遍历数组时需要判断某个数(大一或小一)是否存在=>用哈希表(查询O(1))。
num依次插入set,遍历set,对每个num都尝试匹配num+1,num+2...是否存在。最坏还是O(n²)。
所以加上“跳过逻辑”:存在num-1时跳过,也就是只有num为连续序列的第一个数的时候,才开始内层循环匹配。达到了O(n)。
代码:
class Solution {
public:
int longestConsecutive(vector<int>& nums) {
unordered_set<int> num_set(nums.begin(),nums.end());
int longestStreak = 0;
for (int num : num_set) {
if (!num_set.count(num - 1)) { //不是连续序列的第一个数就跳过
int currentNum = num;
int currentStreak = 1;
while (num_set.count(currentNum + 1)) { //内层循环匹配判断是否存在
currentNum++,currentStreak++;
}
longestStreak = max(longestStreak, currentStreak);
}
}
return longestStreak;
}
};