二叉树的下一个结点

二叉树的下一个结点

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

思想
分三种情况：
1.如果给定结点有右孩子，下一个结点是右孩子的最左孩子。
2.如果给定结点无右孩子且为根节点的左孩子，下一个是根节点。
3.如果给定结点无右孩子且为根节点的右孩子，则找到其为左分支的根节点，下一个为根。

代码
（注意边界条件。）

public static 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 pParent = pNode.next;
            TreeLinkNode pCurrent = pNode;
            if (pCurrent == pParent.left)
                pNext = pParent;
            //他是根节点的右孩子，找到他为左分支的根节点
            while (pParent != null && pCurrent == pParent.right) {
                pCurrent = pParent;
                pParent = pParent.next;
            }
            pNext = pParent;
        }
        return pNext;
    }