数组-128. 最长连续序列(20200523)

本题目在字节跳动的面试过程中出现过，所以在此进行总结一下。

题目描述

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

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

示例:

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

题目分析

本题目要求找到最长连续序列的长度，但是数组尚未排序，那么第一个可以想到的解决办法就是先排序，然后找最长连续序列的长度。第二种解法可以利用set的特性，O(1)的时间复杂度就可以取到某个值。

参考代码

• 排序后找到最长连续序列

public int longestConsecutive(int[] nums) {
    int length = nums.length;
    if (length == 0) {
        return length;
    }
    Arrays.sort(nums);
    List<Integer> list = new ArrayList<>();
    for (int i = 0; i < length; i++) {
        list.add(nums[i]);
    }
    int maxLen = 0;
    int currentLen = 1;
    for (int i = 0; i < length; ) {
        int next = nums[i] + 1;
        if (list.contains(next)) {
            currentLen++;
            i = list.indexOf(next);
        } else {
            i++;
            maxLen = Math.max(maxLen, currentLen);
            currentLen = 1;
        }
    }
    maxLen = Math.max(maxLen, currentLen);

    return maxLen;
}

• 利用set来找最长连续序列

public int longestConsecutive3(int[] nums) {
    Set<Integer> num_set = new HashSet<Integer>();
    for (int num : nums) {
        num_set.add(num);
    }

    int longestStreak = 0;

    for (int num : num_set) {
        // 如果某个数字的-1不存在该数组中，那么在内层while循环中必然不会被访问到，反之在内层while循环中一定会被访问到。
        if (!num_set.contains(num - 1)) {
            int currentNum = num;
            int currentStreak = 1;

            while (num_set.contains(currentNum + 1)) {
                currentNum += 1;
                currentStreak += 1;
            }

            longestStreak = Math.max(longestStreak, currentStreak);
        }
    }

    return longestStreak;
}