给定一个二叉树
struct Node {
int val;
Node *left;
Node *right;
Node *next;
}
填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL。
初始状态下,所有 next 指针都被设置为 NULL。
进阶:
你只能使用常量级额外空间。
使用递归解题也符合要求,本题中递归程序占用的栈空间不算做额外的空间复杂度。
解题思路1:
这一题和【LeetCode】116. Populating Next Right Pointers in Each Node很类似,完全可以一样采用队列层序遍历的方法(指向下一个会有一个判断,不能时当前层的最后一个,及1,3,7,他们都直接指向空)
1进队列,1出队列,1指向队列下一个空,1左(2)进队,1右(3)进队
2出队列,2指向队列下一个3,2左(4)进队,2右(5)进队
3出队列,3指向队列下一个空,3右(7)进队
4出队列,4指向队列下一个5,3右(7)进队
队列有457,然后,逐个出队列并指向下一个,最后7指向空,就完事了
提交代码1:(队列层序,Runtime: 52 ms, faster than 64.76 % )
class Solution:
def connect(self, root: 'Node') -> 'Node':
if not root:
return root
Q = collections.deque([root])
while Q:
size = len(Q)
for i in range(size):
node = Q.popleft()
if i < size - 1:
node.next = Q[0]
if node.left:
Q.append(node.left)
if node.right:
Q.append(node.right)
return root
解题思路2:
跟【LeetCode】116. Populating Next Right Pointers in Each Node思路二类似,请参考这个,https://leetcode-cn.com/problems/populating-next-right-pointers-in-each-node-ii/solution/tian-chong-mei-ge-jie-dian-de-xia-yi-ge-you-ce-j-4/
提交代码2:(Runtime: 52 ms, faster than 64.76 % )
class Solution:
def processChild(self, childNode, prev, leftmost):
if childNode:
if prev:
prev.next = childNode
else:
leftmost = childNode
prev = childNode
return prev, leftmost
def connect(self, root: 'Node') -> 'Node':
if not root:
return root
leftmost = root
while leftmost:
prev, curr = None, leftmost
leftmost = None
while curr:
prev, leftmost = self.processChild(curr.left, prev, leftmost)
prev, leftmost = self.processChild(curr.right, prev, leftmost)
curr = curr.next
return root