给定一个完美二叉树,其所有叶子节点都在同一层,每个父节点都有两个子节点。二叉树定义如下:
struct Node { int val; Node *left; Node *right; Node *next; } 填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL。
初始状态下,所有 next 指针都被设置为 NULL。
示例: 解题思路1:
这一题可以采用队列层序遍历的方法(指向下一个会有一个判断,不能时当前层的最后一个,及1,3,7,他们都直接指向空)
1进队列,1出队列,1指向队列下一个空,1左(2)进队,1右(3)进队2出队列,2指向队列下一个3,2左(4)进队,2右(5)进队3出队列,3指向队列下一个空,3左(6)进队,3右(7)进队4出队列,4指向队列下一个5,3左(6)进队,3右(7)进队队列有4567,然后,逐个出队列并指向下一个,最后7指向空,就完事了提交代码1:(队列层序,Runtime: 64 ms, faster than 81.18 % )
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:
这一题不用队列也能做,每次让根节点的左节点指向右节点,如果根节点有下一个指向节点,那么根节点右节点指向根节点下一个节点的左节点 提交代码2:(Runtime: 64 ms, faster than 81.18 % )
class Solution: def connect(self, root: 'Node') -> 'Node': if not root: return root leftmost = root while leftmost.left: head = leftmost while head: head.left.next = head.right if head.next: head.right.next = head.next.left head = head.next leftmost = leftmost.left return root