LeetCode-128.最长连续序列(相关话题:并查集)

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

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

示例:

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

并查集:初始时,让每个元素都作为一个含有单元素的集合存在

然后按照一定顺序处理每个元素,将同一组的元素(此题中即为相邻数字)所在的集合合并

此题中,用Map<Integer, List<Integer>>

key表示每个元素,value为长度为2的数组,第一个元素为每个元素(key)所属的集合标识,第二个元素为同一个集合中的元素个数

遍历所给元素,将相邻元素所在集合合并,同时更新所属集合标识和所属集合中的元素个数

有关并查集的详解可参考博客:https://blog.csdn.net/u013546077/article/details/64509038

Java代码:

public int longestConsecutive(int[] nums) {
    int len = nums.length, maxCnt = 0;
    Map<Integer, List<Integer>> map = new HashMap<>(len);
    for(int i = 0; i < len; i++){
        if (map.containsKey(nums[i]))
            continue;

        List<Integer> col = new ArrayList<>(2);
        col.add(nums[i]);
        col.add(1);
        if (map.containsKey(nums[i]+1)) {
            col.set(1, map.get(nums[i]+1).get(1)+1);
            map.get(nums[i]+1).set(0, nums[i]);
        }
        if (map.containsKey(nums[i]-1)) {
            col.set(0, nums[i]-1);
            int key = nums[i]-1;
            while (key != map.get(key).get(0)){
                map.get(key).set(1, map.get(key).get(1) + col.get(1));
                key = map.get(key).get(0);
            }
            map.get(key).set(1, map.get(key).get(1) + col.get(1));
            maxCnt = Math.max(maxCnt, map.get(key).get(1));
        }
        maxCnt = Math.max(maxCnt, col.get(1));
        map.put(nums[i], col);
    }

    return maxCnt;
}

猜你喜欢

转载自blog.csdn.net/weixin_38823568/article/details/85210607