剑57—二叉树的下一个节点

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

/*
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;

          /*
          (1)如果有右子树，则找右子树的最左节点
          (2)如果没有右子树，则
            a)是父节点左孩子，那么父节点就是下一个节点 ；
            b)是父节点的右孩子,找他的父节点的父节点的父节点...,直到整体上当前节点处在其父节点的左孩子位置。
        */

        //如果有右子树，则找右子树的最左节点
        if(pNode.right!=null)
        {
            pNode=pNode.right;
            while(pNode.left!=null)
                pNode=pNode.left;
            return pNode;
        }

        //如果没右子树，则寻找 当前节点 整体上是处在 父节点左孩子位置的 第一个父节点
        while(pNode.next!=null)//next是指向父节点
        {
            if(pNode.next.left==pNode)//pNode.next.left父节点的左孩子  是pNode
                return pNode.next;//该父节点
            pNode=pNode.next;//否则，继续寻找父节点的父节点
        }

        return null;//如果没有,那么它就是尾节点
    }
}