欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

python [LeetCode] 最长连续序列

程序员文章站 2024-02-25 09:45:46
...

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

要求算法的时间复杂度为 O(n)。

示例

输入: [100, 4, 200, 1, 3, 2]
输出: 4
解释: 最长连续序列是 [1, 2, 3, 4]。它的长度为 4。

思路
这题并不难,需要考虑到以下情况。
一. 当数组为空时,返回长度为0
二. 若遇到序列[0,1,1,2],应该返回长度3而不是4。

我的思路时间复杂度是O(2n),不太符合规范,之后有时间改进一下。
先将数组排序(从小到大),然后遍历数组的元素。若当前值比上一个值大1,则temp+1,若当前值与上一个值相等,则pass(不做操作),否则就返回temp和result的最大值给result,并令temp=1,继续遍历。

代码
68个用例,耗时60ms,击败50% python3提交记录

class Solution:
    def longestConsecutive(self, nums: List[int]) -> int:
        if not nums:   
            return 0
        nums.sort()
        result = 1     #最终结果,初始化为1
        temp = 1       #变量,记录每一段的最长连续序列长度
        length = len(nums)
        for i in range(1,length):
            if (nums[i] -nums[i-1]) == 1:
                temp += 1
            elif (nums[i] -nums[i-1]) == 0:
                pass
            else:
                result = max(result,temp)
                temp = 1
        result = max(result,temp)
        return result