leetcode: 最长连续序列(hash表)

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

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

一刷:

暴力解法的话很简单的想到需要两层循环,时间复杂度为O(n),这里可以利用hash表来存放原数组,将每次循环数组后的遍历操作变为hash索引操作,就可以将原本判断需要O(n)的时间降为O(1)

class Solution:
    def longestConsecutive(self, nums: List[int]) -> int:
        nums_hash = set(nums)
        longest = 0
        current_num = 0
        for num in nums_hash:
            if num - 1 not in nums_hash:
                current = num
                current_num = 1
                while current + 1 in nums_hash:
                    current += 1
                    current_num += 1
                longest = max(longest, current_num)
        return longest

 

你可能感兴趣的:(python)