经典算法题:最长连续序列

题目

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

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

示例:

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

思路

  1. 将所有数装进HashSet
  2. 从数组第一个元素开始查HashSet,不仅要查它前面的,还要查它后面的,比如:3,2,1,4,6,查3,查前面的会查到2,1,查后面的会查到4。一旦一个元素被查到,直接将这个数从Set中移除,这样一来,3,2,1,4直接全被踢出Set;再开始查2,2已经被踢出去了(HashSet中不包含2),可以直接跳过,1,4亦是如此,最后查6。整个过程只需要记录中间最长的即可。

实现

public int longestConsecutive(int[] nums) {
        Set set = new HashSet<>();

        for (int num : nums) {
            set.add(num);
        }

        int max = 0;

        for (int num : nums) {
            if (set.remove(num)) {
                int cur = num;

                int curMax = 1;

                while (set.remove(cur - 1)) {
                    cur--;
                }

                curMax += (num - cur);

                cur = num;

                while (set.remove(cur + 1)) {
                    cur++;
                }

                curMax += (cur - num);

                max = Math.max(max, curMax);
            }
        }

        return max;
    }

 

你可能感兴趣的:(经典算法题:最长连续序列)