leetcode 128. 最长连续序列

题目描述

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

题解

哈希表存储连续区间的长度,遍历数组,如果数已在哈希表中则跳过,若还没有加入,则取出其左右相邻数表示的连续区间长度相加,然后更新左右区间两端点的长度值

示例代码(go)

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
func longestConsecutive(nums []int) int {
res := 0
hash := make(map[int]int)

for i := 0; i < len(nums); i++ {
num := nums[i]
if _, ok := hash[num]; ok {
continue
}
hash[num] += 1
v1, ok1 := hash[num-1]
v2, ok2 := hash[num+1]

if ok1 {
hash[num] += v1
}
if ok2 {
hash[num] += v2
}

if hash[num] > res {
res = hash[num]
}

if ok1 {
hash[num-v1] = hash[num]
}
if ok2 {
hash[num+v2] = hash[num]
}
}

return res
}