填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL
。
初始状态下,所有 next 指针都被设置为 NULL
。
说明:
- 你只能使用额外常数空间。
- 使用递归解题也符合要求,本题中递归程序占用的栈空间不算做额外的空间复杂度。
- 你可以假设它是一个完美二叉树(即所有叶子节点都在同一层,每个父节点都有两个子节点)。
方案一:利用层次遍历,连接每一层的节点
# Definition for binary tree with next pointer.
# class TreeLinkNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
# self.next = None
class Solution:
# @param root, a tree link node
# @return nothing
def connect(self, root):
if not root:
return
stack = [root]
while stack:
l = len(stack)
for i in range(l):
cur = stack.pop(0)
if i < l - 1:
cur.next = stack[0]
if cur.left:
stack.append(cur.left)
if cur.right:
stack.append(cur.right)
return
方案二:递归
class Solution:
# @param root, a tree link node
# @return nothing
def connect(self, root):
if root and root.left: #如果该节点有左右子树
root.left.next = root.right #先将左右子树的节点用next指针链接起来(由于是完美二叉树,所以如果有左子树,肯定有右子树)
if root.next: #如果该节点的next指针指向的节点不为None,则指向next指向的节点的左子树的节点
root.right.next = root.next.left
else:
root.right.next = None
self.connect(root.left)
self.connect(root.right)