57二叉树的下一个结点

题目描述

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

思路分析

总结为
如果一个结点的右子树不为空，下一个结点为右子树的最左结点
否则，向上找第一个左链接指向的树包含该节点的祖先节点

代码实现

public static TreeLinkNode GetNext(TreeLinkNode pNode) {
    if (pNode == null) {
        return null;
    }
    //右子树不为空，下一结点为右子树的最左结点
    if (pNode.right != null) {
        pNode = pNode.right;
        while (pNode.left != null) {
            pNode = pNode.left;
        }
        return pNode;
    }
    //向上找第一个左连接 包含此节点树的节点
    while (pNode.next != null) {
        if (pNode.next.left == pNode) {
            return pNode.next;
        }
        pNode = pNode.next;
    }
    return null;
}