给定一棵二叉查找树,以及某个结点的值。查找该结点的下一个结点。如果该结点是最大的,则返回 null
对于二叉查找树而言,它是中序遍历有序的。某结点的下一个结点 就是:中序遍历输出的下一个结点。
假设需要查找 node 结点的下一个结点,需要考虑三种情况:
①node 节点有右孩子
下一个结点就是以node结点的右孩子为根的子树中的最左下结点。如下图:node=8,它的下一个结点为12.
②node 节点没有右孩子时,node节点是其父结点的左孩子。如下图,结点8的下一个结点是结点12
③node 节点没有右孩子时,node节点是其父结点的右孩子,如下图,结点14 的下一个结点是 结点16
代码
package jichu;
public class Tree {
class Node{
int val;
Node left;
Node right;
Node parent;
}
public class solution{
public Node ChaTree(Node node){
Node cur=null;
//如果该节点有右孩子,右孩子的最左边的叶子节点
if(node.right!=null){
cur=node.right;
while(cur.left!=null){
cur=cur.left;
}
return cur;
}
//如果该节点是父节点的左孩子
else if(node.parent!=null){
if(node==node.parent.left){
return node.parent;
}
}
//向上找父节点,直到父节点是父节点的父节点的左孩子
Node pcur=node.parent;
while(pcur.parent!=null){
if(pcur==pcur.parent.left){
return pcur.parent;
}
//继续向上找父节点
pcur=pcur.parent;
}
return null;
}
}
}