LC128-最长连续序列

83 阅读1分钟

128. 最长连续序列

给定一个未排序的整数数组 nums ,找出数字连续的最长序列(不要求序列元素在原数组中连续)的长度。

请你设计并实现时间复杂度为 O(n) 的算法解决此问题。

示例 1:

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

输入:nums = [0,3,7,2,5,8,4,6,0,1]
输出:9
 

提示:

0 <= nums.length <= 105 -109 <= nums[i] <= 109

解答:

  • 放在TreeMap中,形成有序集合
  • 遍历,统计相邻元素相差为1的连续长度,找出长度最长的
import java.util.Iterator;
import java.util.TreeMap;

public class LongestConsecutive128 {
    public static int longestConsecutive(int[] nums) {
        if (nums == null || nums.length == 0) return 0;
        TreeMap<Integer, Integer> treeMap = new TreeMap<>();
        for (int i = 0; i < nums.length; i++) {
            treeMap.put(nums[i], nums[i]);
        }

        Iterator iterator = treeMap.keySet().iterator();
        int max = 0;
        int curMax = 1;
        while (iterator.hasNext()) {
            int key = (int) iterator.next();
            if (treeMap.higherKey(key) != null && (treeMap.higherKey(key) - key == 1)) {
                curMax++;
                if (max <= curMax) {
                    max = curMax;
                }
            } else {
                if (max <= curMax) {
                    max = curMax;
                }
                curMax = 1;
            }
        }

        return max;
    }
}