128. 最长连续序列

FuzhouJiang2023-08-16 9:17

题目描述

给定一个未排序的整数数组 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

提示：

0 <= nums.length <= 105
-109 <= nums[i] <= 109

解答

cpp 复制代码

class Solution {
public:
    int longestConsecutive(vector<int>& nums) {
        // 将数组先转存哈希表
        // 然后在数组中对每个元素x，在哈希表中查找 x + 1 是否存在

        // 1. 转存哈希表
        unordered_set<int> num_set;
        for(const int &num : nums)
        {
            num_set.insert(num);
        }

        int res = 0;

        for(const int &num : num_set)
        {
            // 应该判断num - 1 是否存在
            // 若存在本轮就不需要计算了
            if(!num_set.count(num-1))
            {
                int cur = num;
                int curlen = 1;

                while(num_set.count(cur + 1))
                {
                    cur += 1;
                    curlen += 1;
                }
                res = max(res, curlen);
            }
        }
        return res;
    }
};