剑指offer(二叉树的下一个节点)

给定一棵二叉查找树，以及某个结点的值。查找该结点的下一个结点。如果该结点是最大的，则返回 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;
      }
   }

}