D-空间代价替代时间代价Map的使用-LeetCode128-最长连续序列

题目

给定一个未排序的整数数组，找出最长连续序列的长度。
要求算法的时间复杂度为 O(n)。
示例:
输入: [100, 4, 200, 1, 3, 2]
输出: 4
解释: 最长连续序列是 [1, 2, 3, 4]。它的长度为 4。

思路

要求复杂度为 O(n),所以采用Map，以空间代价代替时间代价。Map负责存储<nums[i],目前该值所在的最长连续序列长度>。遍历数组，如果是重复的数则不做任何操作。否则，取(nums[i]+1)目前所在的最长连续序列长度，取(nums[i]-1)目前所在的最长连续序列长度，将两者求和再加1得出nums[i]目前该值所在的最长连续序列长度。之后更新map，需要注意的是，只有连续序列的两端才会对之后的数值产生影响，所以只需更新连续序列两端的map值即可。

代码

class Solution {
    public int longestConsecutive(int[] nums) {
        if(nums.length<=1){
            return nums.length;
        }
        int res=0;
        //要求复杂度为 O(n),所以采用Map，以空间代价代替时间代价。
        //map存储<nums[i],目前该值所在的最长连续序列长度>
        Map<Integer,Integer> map=new HashMap<Integer,Integer>();
        for(int i=0;i<nums.length;i++){
            if(map.get(nums[i])!=null){
                continue;
            }
            //取(nums[i]+1)的value
            int after=0;
            if(map.get((nums[i]+1))==null){
                
            }else{
                after=map.get((nums[i]+1));
            }
            //取(nums[i]-1)的value
            int before=0;
            if(map.get((nums[i]-1))==null){
                
            }else{
                before=map.get((nums[i]-1));
            }
            //总和
            int sum=before+1+after;
            if(sum>res){
                res=sum;
            }
            
            map.put(nums[i],sum);
            //只有连续序列的两端才会对之后的数值产生影响，所以只需更新连续序列两端的map值即可。
            map.put(nums[i]+after,sum);
            map.put(nums[i]-before,sum);
        }
        return res;
    }
}

注意

采用Map，以空间代价代替时间代价。