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

最新推荐文章于 2021-02-20 17:51:39 发布

ShawDa

最新推荐文章于 2021-02-20 17:51:39 发布

阅读量939

点赞数

分类专栏： LeetCode

本文链接：https://blog.csdn.net/sinat_36811967/article/details/88406431

版权

LeetCode 专栏收录该内容

240 篇文章 1 订阅

订阅专栏

给定一个完美二叉树，需要填充每个节点的next指针，使其指向右侧相邻节点。若无右侧节点，则设为NULL。递归方法可用于解决此问题，注意处理子节点的情况。

摘要由CSDN通过智能技术生成

给定一个完美二叉树，其所有叶子节点都在同一层，每个父节点都有两个子节点。二叉树定义如下：
struct Node {
int val;
Node *left;
Node *right;
Node *next;
}
填充它的每个 next 指针，让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点，则将 next 指针设置为 NULL。
初始状态下，所有 next 指针都被设置为 NULL。

提示：
你只能使用常量级额外空间。
使用递归解题也符合要求，本题中递归程序占用的栈空间不算做额外的空间复杂度。

递归，注意要判断子节点：

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


class Solution:
    def connect(self, root: 'Node') -> 'Node':
        if not root:
            return None
        if root.left:
            root.left.next = root.right  # 先连接兄弟节点
            if root.next:
                root.right.next = root.next.left
            self.connect(root.left)
            self.connect(root.right)
        return root