LeetCode 128.Longest Consecutive Sequence (最长连续序列)

题目描述:

给定一个未排序的整数数组,找出最长连续序列的长度。

要求算法的时间复杂度为 O(n)

示例:

输入: [100, 4, 200, 1, 3, 2]
输出: 4
解释: 最长连续序列是 [1, 2, 3, 4]。它的长度为 4。

AC C++ Solution:

方法1:

利用现成的STL标准库算法,先进行sort()排序,然后unique()删除重复元素。最后判断相邻元素是否连续。

不过这种方案虽然简单,但时间复杂度比较高,执行用时为20ms。

代码:

class Solution {
public:
    int longestConsecutive(vector<int>& nums) {
        int n = nums.size();
        if(n < 2)
            return n;
        
        sort(nums.begin(),nums.end());
        unique(nums.begin(),nums.end());
        
        int cnt = 1;
        int maxl = 1;
        
        for(int i = 1; i < n; i++) {
            if(nums[i] - nums[i-1] == 1) {
                cnt++;
                maxl = max(cnt,maxl);
            }
            else
                cnt = 1;
        }
        
        return maxl;
    }
};

方法2:

使用unordered_set容器(此处不使用set容器,因为set是默认有序的,初始化时耗时多),

对于数组中的每个元素,在set中查找相邻的数字是否存在。

class Solution {
public:
    int longestConsecutive(vector<int> &num) {
        if(num.empty())
            return 0;
        unordered_set<int> record(num.begin(),num.end());
        int res = 1;
        for(int n : num){
            if(record.find(n)==record.end()) continue;
            record.erase(n);
            int prev = n-1,next = n+1;
            while(record.find(prev)!=record.end()) record.erase(prev--);        //查找一次后就删除,避免重复查找
            while(record.find(next)!=record.end()) record.erase(next++);
            res = max(res,next-prev-1);
        }
        return res;
    }
};

猜你喜欢

转载自blog.csdn.net/amoscykl/article/details/82983806