128. 最长连续序列
https://leetcode-cn.com/problems/longest-consecutive-sequence/
解法一:
用哈希表存储每个端点值对应连续区间的长度
若数已在哈希表中:跳过不做处理
若是新数加入: 取出其左右相邻数已有的连续区间长度 left 和 right
计算当前数的区间长度为:cur_length = left + right + 1
根据 cur_length 更新最大长度 max_length 的值
更新区间两端点的长度值
class Solution:
def longestConsecutive(self, nums: List[int]) -> int:
hash_dict = {}
max_len = 0
for num in nums:
if num not in hash_dict:
left = hash_dict.get(num-1, 0) #左边区间长度,不存在则为0
right = hash_dict.get(num+1, 0) #右边区间长度,不存在则为0
cur_len = left + right + 1
max_len = max(max_len, cur_len)
#更新区间两端点
hash_dict[num] = cur_len #注意
hash_dict[num - left] = cur_len
hash_dict[num + right] = cur_len
#print(hash_dict, num, left, right)
return max_len
二、并查集
“连续”可以理解为num与num+1的连通关系,巧妙转化为并查集问题
对每个数num,检查num+1是否存在,若存在将他们连通。
并查集的规模为n,每个节点为num的下标i(其实节点值取什么无所谓,主要是为了合并,只要不重合即可,取num也可,但num的范围太大且不一定连续,所以取i较合适)
将每个数存入map,key为num,值为i。然后遍历map。(为何不是遍历nums?因为nums可能有重复数字)
class UnionFind:
# 构造函数传入totalNodes为总节点数
def __init__(self, totalNodes):
# parents[i]表示i的根,初始i的根为其自身
self.parents = [i for i in range(totalNodes)]
# 连通分量数
self.count = totalNodes
# 树的“重量”
self.size = [1] * totalNodes
# 合并连通区域是通过find来操作的, 即看这两个节点是不是在一个连通区域内.
def union(self, node1, node2):
root1 = self.find(node1)
root2 = self.find(node2)
# 不连通才合并
if root1 != root2:
# 小树合并到大树
if self.size[root1] > self.size[root2]:
self.parents[root2] = root1
self.size[root1] += self.size[root2]
else:
self.parents[root1] = root2
self.size[root2] += self.size[root1]
# 连通分量数减一
self.count -= 1
# 查找最终的根
def find(self, node):
while self.parents[node] != node:
self.parents[node] = self.parents[self.parents[node]]
node = self.parents[node]
return node
# 判断两个点是否连通
def isConnected(self, node1, node2):
return self.find(node1) == self.find(node2)
# 返回连通分量个数
def count(self):
return self.count
class Solution:
def longestConsecutive(self, nums: List[int]) -> int:
n = len(nums)
if n == 0: return 0
uf = UnionFind(n)
map = {}
for i in range(n):
map[nums[i]] = i
#若num和num+1都存在, 将他们连通
for num in map:
if num+1 in map:
uf.union(map.get(num), map.get(num+1))
#求连通分量的最大值
res = 0
for size in uf.size:
res = max(res, size)
return res
最后更新于