Leetcode #128. Longest Consecutive Sequence

最新推荐文章于 2025-11-30 19:10:29 发布

原创最新推荐文章于 2025-11-30 19:10:29 发布 · 207 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #python #hard #array

Leetcode 专栏收录该内容

16 篇文章

订阅专栏

本文介绍了一种寻找未排序整数数组中最长连续元素序列的算法，并提供了具体实现方案。该算法采用字典来判断连续性，实现了O(n)的时间复杂度。

题目

Given an unsorted array of integers, find the length of the longest consecutive elements sequence.

Your algorithm should run in O(n) complexity.

Example:

Input: [100, 4, 200, 1, 3, 2]
Output: 4
Explanation: The longest consecutive elements sequence is [1, 2, 3, 4]. Therefore its length is 4.

题意

给出不加以修整的数组，找出里面最长的元素列，元素序列可以隔开然后排序是乱序

想法

当时第一个想法是建树，利用树结构进行排序，然后把树进行裁剪成多棵树，但是这样的实现会比较麻烦，而且这个复杂度会比较高，肯定不止O(n)，但是最主要想法就是要能够判断连续性，所以后面就想着利用字典实现搜索，字典搜索的复杂度是O(1)。
由于是乱序，只记录元素的种数，不考虑个数，所以遍历的时候只用对同一个元素遍历一次，然后通过一个数组记录从当前数字出发能往后连续的最长长度，实现就是不停的尝试下一个，如果不行就停止记录当前长度。有点递推式的感觉。
值得注意的是当时这道题直接交的时候排50%几，看见前面的实现方法是类似于直接往前往后找，边找边丢，想法也是很直观，后来我试了一下直接先用set过滤一遍元素然后用list重构，排名就到93%了。第一道hard。

实现参考

class Solution:
    def longestConsecutive(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        cnts = dict()
        visited = dict()

        nums = list(set(nums))

        for num in nums:
            cnts[num]=1
            visited[num] = -1

        maxlen = 0
        for num in nums:
            if visited[num] == -1:
                print(num)
                tmplen = cnts[num]
                tmpnum = num
                while tmpnum+1 in cnts and visited[tmpnum]==-1:
                    tmplen += cnts[tmpnum+1]
                    visited[tmpnum]=1
                    tmpnum=tmpnum+1
                visited[num] = 1
                cnts[num] = tmplen
                if tmplen>maxlen:
                    maxlen = tmplen
        return maxlen