[LintCode 124] 最长连续序列(Python)

问题描述

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

说明

样例

给出数组[100, 4, 200, 1, 3, 2]，这个最长的连续序列是 [1, 2, 3, 4]，返回所求长度 4

思路

用一个字典存储中间值。遍历数组，对于数字i，找到i-1和i+1对应的value值,如果不存在则记0。然后把i的value值设为i-1,i+1的value值之和，并加1，相当于连接起来。同时置最左端和最右端的数的value值为i的value值（中间的数都已经出现过，不会再用到了）。然后更新一次最大值。

代码

class Solution:
"""
@param: num: A list of integers
@return: An integer
"""
def longestConsecutive(self, num):
# write your code here
if num is None or len(num) == 0:
return 0
m = {}
res = 0
for i in num:
if i not in m:
l = 0
r = 0
if i - 1 in m:
l = m[i - 1]
if i + 1 in m:
r = m[i + 1]
m[i] = 1 + r + l
m[i + r] = 1 + r + l
m[i - l] = 1 + r + l
res = max(res, m[i])
return res

复杂度分析

时间复杂度O(n)，空间复杂度O(n)。