题目
给定一个未排序的整数数组 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
分析: 如果是先排序,再对排序好的数组遍历的话时间复杂度是
O
(
n
l
o
g
(
n
)
)
O(nlog(n))
O(nlog(n)), 要去用
O
(
n
)
O(n)
O(n)的时间复杂度完成,因此要用到哈希表,哈希表查找的平均时间复杂度是
O
(
1
)
O(1)
O(1), 先用哈希表存储数组中的元素,set去重,然后遍历表中的元素,判断该元素往上的元素是否在表中,比如当前元素是5, 则判断5、6、7…是否存在表中,知道找到一个不存在的数位置,记录长度,同时继续对表中数往后遍历。如果单纯这么做,那么时间复杂度是
O
(
n
2
)
O(n^2)
O(n2), 因为表中每一个元素都需要进行连续性判断,存在重叠问题,如果num和num+1均在表中,那么最长序列必然是num为起始,因此,只需要判断num-1是否在表中,如果在表中,则该num可以直接跳过,因为最终遍历到num-1的时候,序列必然包含num。这样就能把时间复杂度降低到
O
(
n
)
O(n)
O(n)
总结:
- 创建一个哈希表,存储数组中去重的元素
- 对表中元素遍历,判断num-1是否在表中
- num-1在表中,则直接跳过,继续遍历表中后面的元素
- num-1不在表中,那么序列的起点为num, 判断num+1, num+2, …是否在表中,找到存在的最长序列
class Solution:
def longestConsecutive(self, nums: List[int]) -> int:
nums_set = set(nums)
max_length = 0
for num in nums_set:
cur_length = 1
cur_num = num
if cur_num-1 in nums_set:
continue
else:
while cur_num+1 in nums_set:
cur_num += 1
cur_length += 1
max_length = max(cur_length, max_length)
return max_length