给定一个未排序的整数数组,找出最长连续序列的长度。
要求算法的时间复杂度为 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;
}