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