117. 填充每个节点的下一个右侧节点指针 II

最新推荐文章于 2024-07-13 23:51:10 发布

追梦少年_duang

最新推荐文章于 2024-07-13 23:51:10 发布

阅读量27

点赞数

分类专栏：数据结构与算法文章标签： python 算法

本文链接：https://blog.csdn.net/qq_45746957/article/details/131926278

版权

数据结构与算法专栏收录该内容

2 篇文章 0 订阅

订阅专栏

树的层序遍历，使用前继指针。

"""
# Definition for a Node.
class Node:
    def __init__(self, val: int = 0, left: 'Node' = None, right: 'Node' = None, next: 'Node' = None):
        self.val = val
        self.left = left
        self.right = right
        self.next = next
"""

class Solution:
    def connect(self, root: 'Node') -> 'Node':
        if root is None:
            return None

        queue = collections.deque([root])
        while queue:
            size = len(queue)
            prev = None
            for _ in range(size):
                node = queue.popleft()
                if prev:
                    prev.next = node
                prev = node
                left, right = node.left, node.right
                if left:
                    queue.append(left)
                if right:
                    queue.append(right)
        return root