最长连续序列

问题描述

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

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

示例:

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

 

import java.util.HashMap;
import java.lang.Math;

class Solution {
    public int longestConsecutive(int[] nums) {
        if(null == nums || nums.length == 0){
            return 0;
        }
        
        int max = 0;
        HashMap<Integer,Integer> map = new HashMap<>();
        for (int i = 0; i < nums.length; i++){
            if(map.getOrDefault(nums[i], 0) == 0){
                int left = map.getOrDefault(nums[i] - 1, 0);
                int right = map.getOrDefault(nums[i] + 1, 0);
                
                int length = left + right + 1;
                map.put(nums[i], length);
                if(left != 0){
                    map.put(nums[i] - left, length);
                }
                if(right != 0){
                    map.put(nums[i] + right, length);
                }

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

        return max;
    }
}