算法分析与设计,第18周博客
128. Longest Consecutive Sequence
Given an unsorted array of integers, find the length of the longest consecutive elements sequence.
For example,
Given [100, 4, 200, 1, 3, 2]
,
The longest consecutive elements sequence is [1, 2, 3, 4]
. Return its length: 4
.
Your algorithm should run in O(n) complexity.
给定一组无序数,要求其中的最长连续序列的长度,而且时间复杂度要求是O(n)。
如果不指定时间复杂度的话,那么最容易想到的就是先排序了,然后在遍历整个数组一次,但这样的时间复杂度是O(n*log(n))。不符合题目的要求,所以,排序方法是不行的。时间复杂度要求是O(n),这也就要求在常数级别的遍历中就把答案找出来。那么这样就必须保存下一些信息,也就是,对于每个n[i],都能够记录下与之相关的最长连续序列长度,那么对于剩下的数,我们只需要查看是否有与n[i]相连接的,如果有,就把两者的长度拼起来,否则,设为1.
总结如下:对于每个数p,找出p-1和p+1的 最长连续序列长度,设分别为left和right,那么p的最长连续序列长度就是left+1+right;另外,需要更新左右两端的最长连续序列长度,把它们的值都设为left+1+right.
这里选择了使用无序字典来存储信息,在c++里,无序字典是根据hash值来进行存储和查找的,平均时间复杂度是O(1),所以代码的总体时间复杂度也是O(n),代码如下:
class Solution {
public:
int longestConsecutive(vector<int>& nums) {
int n = nums.size();
if (n == 0)
return n;
unordered_map<int, int> map;
int longest = 1;
for (int i = 0; i < n; ++i) {
if (map.find(nums[i]) != map.end())
continue;
int left = map.find(nums[i]-1) == map.end() ? 0 : map.find(nums[i]-1)->second;
int right = map.find(nums[i]+1) == map.end() ? 0 : map.find(nums[i]+1)->second;
int cur = left+1+right;
map[nums[i]] = cur;
map[nums[i]-left] = cur;
map[nums[i]+right] = cur;
longest = max(longest, cur);
}
return longest;
}
};