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

最新推荐文章于 2022-03-18 11:05:17 发布

LotusQ

最新推荐文章于 2022-03-18 11:05:17 发布

阅读量132

点赞数

分类专栏：面试之路

本文链接：https://blog.csdn.net/qq_30057549/article/details/103506955

版权

面试之路专栏收录该内容

122 篇文章 7 订阅

订阅专栏

难度：中等
题目描述：
在这里插入图片描述

思路总结：一种通用的解法，层次遍历设置pre指针。注意不要忘了让pre往后移动。
题解一：

"""
# 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
"""
from collections import deque
class Solution:
    def connect(self, root: 'Node') -> 'Node':
        #层次遍历
        if not root:return None
        pre = None
        queue = deque([root])
        while queue:
            n = len(queue)
            while n > 0:
                cur = queue.popleft()
                if pre:
                    pre.next = cur
                    pre = pre.next
                else:
                    pre = cur
                if cur.left:queue.append(cur.left)
                if cur.right:queue.append(cur.right)
                n -= 1
            pre = None
        return root