LeetCode 128. 最长连续序列

128. 最长连续序列

Medium

题目描述

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

解题思路：

• 首先，把数组中的所有元素放入一个哈希集合中，方便后面的查找。
• 然后，遍历哈希集合中的每个元素，如果该元素的前一个元素不在集合中，说明它是一个连续序列的起点。
• 接着，从该元素开始，向后查找是否有连续的元素在集合中，记录当前序列的长度，并更新最长序列的长度。
• 最后，返回最长序列的长度。

代码实现：

#include<iostream>
#include<unordered_set>
#include<vector>
using namespace std;

class Solution {
public:
    int longestConsecutive(vector<int>& nums) {
        unordered_set<int> num_set; // 哈希集合，用来存储数组中的元素
        for (const int& num : nums) {
            num_set.insert(num); // 把数组中的每个元素放入集合中
        }

        int longestStreak = 0; // 最长连续序列的长度

        for (const int& num : num_set) { // 遍历集合中的每个元素
            if (!num_set.count(num - 1)) { // 如果该元素的前一个元素不在集合中，说明它是一个连续序列的起点
                int currentNum = num; // 当前元素
                int currentStreak = 1; // 当前连续序列的长度

                while (num_set.count(currentNum + 1)) { // 如果当前元素的后一个元素在集合中，说明序列还没有结束
                    currentNum += 1; // 更新当前元素为后一个元素
                    currentStreak += 1; // 更新当前序列的长度
                }

                longestStreak = max(longestStreak, currentStreak); // 更新最长序列的长度
            }
        }

        return longestStreak; // 返回最长序列的长度           
    }
};


int main() {
    Solution sol; // 创建一个Solution对象
    vector<int> nums; // 创建一个空的数组
    int n; // 用来存储数组的大小
    cout << "请输入数组的大小：" << endl;
    cin >> n; // 输入数组的大小
    cout << "请输入数组的元素：" << endl;
    for (int i = 0; i < n; i++) {
        int x; // 用来存储每个元素
        cin >> x; // 输入每个元素
        nums.push_back(x); // 把元素放入数组中
    }
    int ans = sol.longestConsecutive(nums); // 调用算法函数，得到结果
    cout << "最长连续序列的长度是：" << ans << endl; // 输出结果
    return 0;
}