给定一个二叉树
struct Node {
int val;
Node *left;
Node *right;
Node *next;
}
填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL。
初始状态下,所有 next 指针都被设置为 NULL。
进阶:
你只能使用常量级额外空间。
使用递归解题也符合要求,本题中递归程序占用的栈空间不算做额外的空间复杂度。
示例:
输入:root = [1,2,3,4,5,null,7]
输出:[1,#,2,3,#,4,5,7,#]
解释:给定二叉树如图 A 所示,你的函数应该填充它的每个 next 指针,以指向其下一个右侧节点,如图 B 所示。序列化输出按层序遍历顺序(由 next 指针连接),'#' 表示每层的末尾。
提示:
树中的节点数小于 6000
-100 <= node.val <= 100
链接:https://leetcode-cn.com/leetbook/read/data-structure-binary-tree/xorvud/
来源:力扣(LeetCode)
# Definition for a Node.
class Node:
def __init__(self, val: int = 0, left=None, right=None, next=None):
self.val = val
self.left = left
self.right = right
self.next = next
class Solution:
def connect(self, root):
if not root or (not root.left and not root.right):
return root
def findCousin(node, parent):
tmp = parent.next
while tmp:
if tmp.left:
node.next = tmp.left
break
elif tmp.right:
node.next = tmp.right
break
tmp = tmp.next
if root.left and root.right:
root.left.next = root.right
findCousin(root.right, root)
elif root.left:
findCousin(root.left, root)
elif root.right:
findCousin(root.right, root)
# 下面right和left的顺序不能变;
self.connect(root.right)
self.connect(root.left)
return root