二叉树的下一个结点

题目描述

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

算法解析：
如果当前结点的右子节点不为空，那么下一个结点一定出现在当前结点的右子树上，转换为寻找当前右子树中序遍历的第一个结点。如果当前结点的父节点不为空，且当前结点是父节点的左子节点，则下一个结点一定是父节点，如果当前结点是父节点的右子节点，那么将当前结点和父节点视为一个子树，直到找到一个父节点，以之为左子树。

代码如下：

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 && pCurrent == pParent.right){
                pCurrent = pParent;
                pParent = pParent.next;
            }
            pNext = pParent;
        }
        return pNext;
    }