欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

LeetCode 最长连续序列

程序员文章站 2022-03-11 13:33:15
...

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

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

示例:

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

维护一个hash,key为数组元素,value为以key为端点的连续队列长度。新增元素x时,如果x+1存在,则x+1的value是x之后的连续队列的长度,如果x-1存在,则x-1的value是x之前的连续队列的长度,只需要将x加入进去,然后更新新连续队列的两端元素的长度即可,队列中间元素的长度值可以不用管。

public static int longestConsecutive(int[] nums) {
        int max = 1;
        Map<Integer, Integer> map = new HashMap(nums.length);
        for(int i: nums){
            if(!map.containsKey(i)){
                int right = map.getOrDefault(i + 1, 0);
                int left = map.getOrDefault(i - 1, 0);
                int len = left + right + 1;
                map.put(i, len);
                map.put(i - left, len);
                map.put(i + right, len);
                max = max >= len ? max : len;
            }
        }
        return max;
    }