leetcode(填充每个节点的下一个右侧节点指针 II) - 详解

给定一个二叉树:

struct Node {
  int val;
  Node *left;
  Node *right;
  Node *next;
}

填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL 。

初始状态下,所有 next 指针都被设置为 NULL 。

思路:

采用层级遍历,也可以称广度优先遍历(BFS),建立一个队列存储,按每一层搜索查看

首先建立数组,将根节点放入,开始操作

计算此节点算在的这一层元素个数(队列长度),

对每一层,设一个pre节点初始化为null

for循环重复直到这一层元素均被遍历,使下一层它们的左右子结点均被加入。具体操作是

弹出首个节点,判断pre是否为null。如果为null表明此元素是这一层的首个元素,不需要指向。否则,设置他的next。判断完成后,将他的左右子结点(非空)加入队列。

/*
// Definition for a Node.
class Node {
public int val;
public Node left;
public Node right;
public Node next;
public Node() {}
public Node(int _val) {
val = _val;
}
public Node(int _val, Node _left, Node _right, Node _next) {
val = _val;
left = _left;
right = _right;
next = _next;
}
};
*/
class Solution {
public Node connect(Node root) {
if(root==null)
return null;
LinkedList queue = new LinkedList<>();
queue.add(root);
while(!queue.isEmpty()) {
int levelnum=queue.size();
Node prev=null;
for(int i=0;i

posted on 2025-09-24 19:28  slgkaifa  阅读(9)  评论(0)    收藏  举报

导航