思路如下:
分析二叉树,有三种情况:
1、最右边的结点指向null
2、同一父节点的两个子节点由左指向右
3、不是同一父节点的两个相邻子节点由左指向右
由此可得到三种递归方法,分情况进行调用
/*
// 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;
}
bfs1(root);
if(root.left!=null){
bfs2(root);
if(root.left.left!=null){
bfs3(root.left,root.right);
}
}
return root;
}
public void bfs1(Node root){
//最右边的结点指向空,并对root的right结点继续递归
if(root!=null){
root.next=null;
}
if(root.right!=null){
bfs1(root.right);
}
}
public void bfs2(Node root){
if(root.left!=null){
//同一个父节点,由左指向右
root.left.next=root.right;
}
//继续对两个结点进行递归
if(root.left.left!=null){
bfs2(root.left);
bfs2(root.right);
}
}
public void bfs3(Node current,Node next){
//不同的父节点,由左指向右
if(current.left!=null){
current.right.next=next.left;
}
if(current.left.left!=null){
//子树的上一个根节点相同
bfs3(current.left,current.right);
bfs3(next.left,next.right);
//子树的上一个根节点不同
bfs3(current.right,next.left);
}
}
}
原题地址:
116. 填充每个节点的下一个右侧节点指针