二叉树的下一个结点

---

题目描述

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

---

思路

① 将给出的二叉树中序遍历的结果保存，在找到目标结点Target的下一个结点
② 结点中包含指向父节点的指针，利用该父指针分析到Target结点的位置上情况（不使用额外空间）
Target结点：
存在右子树： Target结点的右子树最左节点就是Next结点（图中 D B E A C G）
不存在右子树：
Target结点为父节点的左孩子：父节点就是Next结点（图中 N L I）
Target结点为父节点的右孩子：父节点的父节点的…直到当前结点时其父节点的左孩子（图中HJKM）

---

具体图解如下（注：这里引用牛客网***小河沟大河沟***的图解）

---

具体实现代码如下

public class Solution {
    public TreeLinkNode GetNext(TreeLinkNode pNode){
        // 代码的鲁棒性
        if(pNode == null){
            return pNode;
        }
        // 右子树不为空
        if(pNode.right != null){
            return getLeftMost(pNode.right);
        }else{
            TreeLinkNode parent = pNode.next;
            while(parent != null && parent.left != pNode){
                pNode = parent;
                parent = pNode.next;
            }
            return parent;
        }
    }
        // 获取最左节点
    public TreeLinkNode getLeftMost(TreeLinkNode pNode){
        if(pNode == null){
            return pNode;
        }
        while(pNode.left != null){
            pNode = pNode.left;
        }
        return pNode;
    }
}

NowCoder（Online Coding, Please Click）