给定一个 完美二叉树 ,其所有叶子节点都在同一层,每个父节点都有两个子节点。二叉树定义如下:
struct Node {
int val;
Node *left;
Node *right;
Node *next;
}
填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL。
初始状态下,所有 next 指针都被设置为 NULL。
方法一:层序遍历
典型层序遍历步骤,只不过在将该节点左右子树推入队列之前将每一层遍历节点进行连接。
class Solution {
public Node connect(Node root) {
if(root == null)return null;
LinkedList<Node> queue = new LinkedList<>();
queue.offer(root);
while(!queue.isEmpty()) {
int size = queue.size();
Node temp = queue.get(0);
for(int i=1;i<size;i++) { //每层节点首尾相连
temp.next = queue.get(i);
temp = temp.next;
}
for(int i=0;i<size;i++) {
Node temp2 = queue.poll();
if(temp2.left != null)queue.offer(temp2.left);
if(temp2.right != null)queue.offer(temp2.right);
}
}
return root;
}
}
方法二:迭代法
class Solution {
public Node connect(Node root) {
if(root == null)return null;
Node pre = root;
while(pre.left != null) {
Node temp = pre;
while(temp != null) {
temp.left.next = temp.right;
if(temp.next != null) {
temp.right.next = temp.next.left;
}
temp = temp.next;
}
pre = pre.left;
}
return root;
}
}
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/populating-next-right-pointers-in-each-node