二叉树的下一个节点

题目描述

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

/*
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 pNext = null;
        //如果该结点有右子树，则下一个结点是右子树的最左子结点
        if(pNode.right != null){
            TreeLinkNode pRight = pNode.right;
            while(pRight.left != null){
                pRight = pRight.left;
            }
            pNext = pRight;
        }else if(pNode.next != null){
            //如果该结点没有右子树，此时分为两种情况
            //一，如果该结点是父结点的左子结点，那么下一个结点就直接是父节点
            //二，如果该结点是父节点的右子结点，那么一直往上寻找父节点，直到找到一个结点，
            //他是他父节点左子结点的结点，那么下一个节点就是该结点的父节点
            TreeLinkNode pCurrent = pNode; //当前结点
            TreeLinkNode pParent = pNode.next; //父节点
            while(pParent != null && pParent.right == pCurrent){
                //父节点不为空，或者父节点的右结点等于当前结点
                pCurrent = pParent;
                pParent = pCurrent.next;
            }
            pNext = pParent;
        }
        return pNext;
    }
}