给定一个未排序的整数数组 nums ,找出数字连续的最长序列(不要求序列元素在原数组中连续)的长度。
进阶:你可以设计并实现时间复杂度为 O(n) 的解决方案吗?
Given an unsorted array of integers nums, return the length of the longest consecutive elements sequence.
Follow up: Could you implement the O(n) solution?
Example 1:
Input: nums = [100,4,200,1,3,2]
Output: 4
Explanation: The longest consecutive elements sequence is [1, 2, 3, 4]. Therefore its length is 4.
Example 2:
Input: nums = [0,3,7,2,5,8,4,6,0,1]
Output: 9
链接:https://leetcode-cn.com/problems/longest-consecutive-sequence
解决思路:
因为要满足O(n)的时间复杂度,就考虑能否使用哈希表查找来进行处理,因为在哈希表中查找元素是常数级的时间复杂度。
本题是求数组中的数字连续的最长序列,采用map的key来存储vector中的数字,map的value来存储该元素有没有被计算过,初始化的value都为false。然后,遍历数组中的元素,对每个元素ele都在map中循环查找其ele+1和ele-1,直到find的结果为空为止,每找到一次length就递增,并需要将找到元素在map中的value值标记为true,代表该元素已经被计算过了,下次遍历可以直接跳过该元素。最后,我们需要采用longest来记录一下每次求的length中的最长的结果。
代码如下:
class Solution {
public:
int longestConsecutive(vector<int>& nums)
{
unordered_map<int, bool> map;
int longest;
for(auto& ele : nums)
{
map[ele] = false;
}
for(auto& ele : nums)
{
if(map[ele])
{
continue;
}
int length = 1;
for(int j = ele + 1; map.find(j) != map.end(); j++)
{
map[j] = true;
length++;
}
for(int j = ele - 1; map.find(j) != map.end(); j--)
{
map[j] = true;
length++;
}
longest = length > longest ? length : longest;
}
return longest;
}
};