659. 分割数组为连续子序列
给你一个按升序排序的整数数组 num(可能包含重复数字),请你将它们分割成一个或多个子序列,其中每个子序列都由连续整数组成且长度至少为 3 。
如果可以完成上述分割,则返回 true ;否则,返回 false 。
这道题看起来不难,但其实不是很容易有较好的思路。
思路:最小堆+哈希表暂存
遍历每一个数组中的数字,判断该数字 - 1 是否在哈希表中有存储(哈希表中存储了每一个数所对应的一个维护好的最小堆,最小堆存的是在此之前的所有长度),如果有存储,则将该最小堆的堆顶值移动到后一位置,并且长度+1,如果没有,则添加长度为1的最小堆,最后遍历哈希表即可
class Solution:
def isPossible(self, nums: List[int]) -> bool:
ite = collections.defaultdict(list)
for num in nums:
if ite.get(num-1):
Len = heapq.heappop(ite[num-1])
heapq.heappush(ite[num],Len+1)
else:
heapq.heappush(ite[num],1)
return not any(hp and hp[0]<3 for hp in ite.values())
时间复杂度O(nlogn)
空间复杂度O(n)