剑指offer57.二叉树的下一个节点（Java）

题目描述

给定一个二叉树和其中的一个结点，请找出中序遍历顺序的下一个结点并且返回。注意，树中的结点不仅包含左右子结点，同时包含指向父结点的指针。

分析

二叉树中的节点有五种情况，其中序下一个节点为

1. 有右孩子
   当前节点右孩子，一路向左，返回最后一个点

2. 无右孩子，若为双亲的左孩子
   下一个节点为，双亲

3. 无右孩子，若为双亲的右孩子
   分两种情况
   在根节点的左子树上，返回其父节点的父节点
   在根节点的右子树上，返回空

4. 无右孩子，为根节点
   返回空

代码

/*
public class TreeLinkNode {
    int val;
    TreeLinkNode left = null;
    TreeLinkNode right = null;
    TreeLinkNode next = null;

    TreeLinkNode(int val) {
        this.val = val;
    }
}
*/
public class Solution {
    public TreeLinkNode GetNext(TreeLinkNode pNode)
    {
        if(pNode==null)
            return null;
        
        //当前节点
        TreeLinkNode r = pNode;
        
        //右孩子不空，右孩子一路向左
        if(r.right!=null){
            r = r.right;
            while(r.left!=null){
                r = r.left;
            }
            return r;
        }        
       
        //当前节点为根节点
        if(r.next==null)
            return null;
        
       //无右子树，若为双亲的左孩子,下一个节点为父节点
        if(r == r.next.left)
            return r.next;
            
       //为双亲的右孩子，需要判断是在左子树还是右子树中
       //若在左子树，返回父亲的父亲
       if(result(r)){
            return r.next.next;
       }      
        return null;                    
    }
    
    //判断当前节点在根节点的左子树/右子树上
    //判断当前节点的双亲
    public boolean result (TreeLinkNode p){
        //判断当前节点的双亲
        TreeLinkNode root = p;
        TreeLinkNode pre = p; //存储寻根过程中前一个点
        while(root.next!=null){
            pre = root;
            root=root.next;
        }
        
        //判断当前节点在根节点的左子树/右子树上
        if(pre==root.left)
            return true;
        else
            return false;                
    }    
}