最长连续序列
给定一个未排序的整数数组 nums ,找出数字连续的最长序列(不要求序列元素在原数组中连续)的长度。
请你设计并实现时间复杂度为 O(n) 的算法解决此问题。
示例 1:
输入:nums = [100,4,200,1,3,2]
输出:4
解释:最长数字连续序列是 [1, 2, 3, 4]。它的长度为 4。
示例 2:
输入:nums = [0,3,7,2,5,8,4,6,0,1]
输出:9
示例 3:
输入:nums = [1,0,1,2]
输出:3
我用的是map,具体思路是在while
循环中,遍历每个元素,并寻找和它连续的元组,已经找到的元素就置map[nums[i]]
为0,防止重复遍历。
class Solution {
public:
int longestConsecutive(vector<int>& nums) {
int max_len=0;
map<int,int> num_map;
for(int i=0; i<nums.size(); i++) num_map[ nums[i] ]++;
for(auto num : num_map)
{
int start=num.first;
int current=1;
while(num.second && num_map.count(start+1))
{
current+=1;
num_map[start+1]=0;
start++;
}
max_len=max(max_len, current);
}
return max_len;
}
};
但更好的方式还是使用set
,对于每一个num,从num-1开始遍历,以保证所有参与遍历的num 都是当前的连续数组的最小值,这与我的代码中num_map[start+1]=0;
有相同效果。
class Solution {
public:
int longestConsecutive(vector<int>& nums) {
set<int> num_set(nums.begin(), nums.end());
int max_len = 0;
for (int num : num_set)
{
if (num_set.find(num - 1) == num_set.end())
{
int current_num = num;
int current_len = 1;
while (num_set.find(current_num + 1) != num_set.end())
{
current_num++;
current_len++;
}
max_len = max(max_len, current_len);
}
}
return max_len;
}
};