LeetCode 128. 最长连续序列

128. 最长连续序列 - 力扣(LeetCode)

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

示例 3:

复制代码
输入:nums = [1,0,1,2]
输出:3

提示:

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

解题思路:

对于 nums 中的元素 n,以 n 为起点,不断查找下一个数 n+1, n+2, ⋯ 是否在 nums 中,并统计序列的长度。

为了满足O(n)的时间复杂度,

  1. 把 nums 的元素放在 Set 中去重后, O(1) 复杂度判断 n + 1...是否在 nums 中;

  2. 如果 n - 1 在 Set 集合中, 则不能以 n 为起点, 因为 n - 1 为起点的序列一定比 n 为起点的序列长。

java 复制代码
class Solution {
    public int longestConsecutive(int[] nums) {
        // hash
        // Time: O(n)
        // Space: O(n)
        Set<Integer> set = new HashSet<>();
        for (int num : nums) {
            set.add(num);
        }
        int longestStreak = 0;
        for (int num : set) {
            if (!set.contains(num - 1)) {
                int curNum = num;
                int curStreak = 1;
                while (set.contains(curNum + 1)) {
                    curNum++;
                    curStreak++;
                }
                longestStreak = Math.max(longestStreak, curStreak);
            }
        }
        return longestStreak;
    }
}