题目
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