0116. 填充每个节点的下一个右侧节点指针
小于 1 分钟
0116. 填充每个节点的下一个右侧节点指针
- 标签:树、深度优先搜索、广度优先搜索、链表、二叉树
- 难度:中等
题目大意
给定一个完美二叉树,所有叶子节点都在同一层,每个父节点都有两个子节点。二叉树结构如下:
struct Node {
int val;
Node *left;
Node *right;
Node *next;
}
要求填充每个 next 指针,是的这个指针指向下一个右侧节点。如果找不到下一个右侧节点,则将 next 置为 None。
示例:

解题思路
层次遍历。在层次遍历的过程中,依次取出每一层的节点,并进行连接。然后再扩展下一层节点。
代码
import collections
class Solution:
def connect(self, root: 'Node') -> 'Node':
if not root:
return root
queue = collections.deque()
queue.append(root)
while queue:
size = len(queue)
for i in range(size):
node = queue.popleft()
if i < size - 1:
node.next = queue[0]
if node.left:
queue.append(node.left)
if node.right:
queue.append(node.right)
return root