LeetCode—— Longest Consecutive Sequence(128)

最新推荐文章于 2021-10-25 17:26:13 发布

原创最新推荐文章于 2021-10-25 17:26:13 发布 · 196 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

24 篇文章

订阅专栏

本文介绍了一种在未排序整数数组中寻找最长连续元素序列的算法，通过使用哈希表优化搜索过程，确保算法复杂度为O(n)，并提供了解决方案的详细代码实现。

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.

分析：

如果我们要找最长连续序列，以3为例，我们会从两个方向找，即3+1和3-1两个方向，一直延伸下去直到找不到为止，1<-2<-3->4,这样我们如果从3开始，就得到3的连续最长序列是4。同样的，如果我们找4的最长连续序列，1<-2<-3<-4，4的最长连续序列也是4，可以发现，我们上面进行了重复的计算，因为1，2，3，4这个连续序列里，每个数的最长连续序列大小都等于4。于是我们可以做出优化，用哈希表存储所有值，如果该数在最长连续序列中，则从哈希表里删除，避免重复计算。

代码：

class Solution {
public:
    int longestConsecutive(vector<int>& nums) {
        if(nums.empty())
            return 0;
        unordered_set<int> s(nums.begin(), nums.end());
        int max=1;
        for(int n:nums)
        {
            if(s.find(n)==s.end()) continue;
            s.erase(n);
            int pre=n-1;
            int next=n+1;
            int sum=1;
            while(s.find(pre)!=s.end())
            {
                ++sum;
                s.erase(pre);
                --pre;
            }
            while(s.find(next)!=s.end())
            {
                ++sum;
                s.erase(next);
                ++next;
            }
            max = max>sum?max:sum;
        }
        return max;
    }
};