[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)。