给定一个 完美二叉树 ,
其所有叶子节点都在同一层,每个父节点都有两个子节点。
填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL。
初始状态下,所有 next 指针都被设置为 NULL。
思路:因为是完全二叉树,每一层最右边的节点的next指针肯定指向的是next,左边节点的next指针指向根节点的右节点。还有一个需要考虑的问题就是左子树的右子节点如何找到右子树的左子节点,因为next指针指向的是右子节点,所以就完美解决了
代码实现:
public Node connect(Node root) {
//递归
return conn(root);
}
public Node conn(Node root){
if(root == null){
return null;
}
//如果左子节点不为空,那么左子节点的下一个节点指向的就是右子节点,不用管右子节点是否为空
//所有的最右边的节点的next指针都是指向的是null
if(root.left != null){
root.left.next = root.right;
}
//让左子树的右节点和右子树的左子节点相连
if(root.right != null && root.next != null){
root.right.next = root.next.left;
}
conn(root.left); //递归左子树
conn(root.right); //递归右子树
return root;
}
也可以根据层序遍历,让前面的节点指向后一个节点
只不过特别慢罢了
public Node connect(Node root) {
if(root == null){
return null;
}
Queue<Node> qNode = new LinkedList<>();
qNode.add(root);
while (qNode.size() > 0){
int length = qNode.size();
List<Node> level = new ArrayList<>();
for(int i = 0; i < length ; i ++){
Node node = qNode.remove();
level.add(node);
for (int j = 0; j < level.size() - 1; j++) {
level.get(j).next = level.get(j+1);
}
if (node.left != null){
qNode.add(node.left);
}
if(node.right != null ){
qNode.add(node.right);
}
}
}
return root;
}