【剑指Offer】57、二叉树的下一个结点

【剑指Offer】57、二叉树的下一个结点

题目描述：

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

解题思路：

本题解决起来并不是很困难，主要是分析清楚所有可能的不同情况。对于中序遍历序列来说，遵循“左->根->右”的顺序，在深刻理解中序遍历的基础上，结合一些具体的实例我们不难得出以下结论。

找一个结点在中序遍历序列中的下一个结点共有三种不同的情况：

如果一个结点有右子树，那么它的下一个结点就是它的右子树中最左边的那个结点，也就是说，从它的右子结点出发一直访问左指针，最后就可以找到这个最左结点。
如果一个结点没有右子树，那么需要再进一步考虑，不难知道：如果这个结点是其父结点的左结点，那么根据中序遍历规则，它的下一个结点就是它的父结点。
第三种情况略复杂一些，当一个结点既没有右子树，也不是其父结点的左结点时，我们可以沿着指向父结点的指针一直向上遍历，直到找到一个是它自身的父结点的左孩子的结点，如果这样的结点存在，那么这个结点的父结点就是我们要找的下一个结点。
  举例：

以上图中的树为例，其中序遍历序列是：d,b,h,e,i,a,f,c,g。

对应第一种情况（有右子树的情形），例如，图中结点b的下一个结点是h，结点a的下一个结点是f。

对应第二种情况（没有右子树，但是其父结点的左结点的情形），例如，图中结点d的下一个结点是b，f的下一个结点是c。

对应第三种情况（没有右子树，也不是其父结点的左结点的情形），例如，为了找到结点g的下一个结点，我们沿着指向父结点的指针向上遍历，先到达结点c。由于结点c是父结点a的右结点，我们继续向上遍历到达结点a。由于结点a是树的根结点。它没有父结点。因此结点g没有下一个结点。

编程实现（Java）：

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){
        /*
        思路：三种情况：（1）若给定结点有右子树，则下一个结点是它右子树中最左边的结点
            （2）若该结点没有右子树，并且是父节点的左结点，则下一个结点就是他的父节点
            （3）若该结点没有右子树，并且是父节点的右结点，则应该沿着父节点向上，直到找到的结点是一个左结点。则下一个结点是该左结点的父节点
        */
        if(pNode==null)
            return null;
        TreeLinkNode res=null;
        if(pNode.right!=null){ //有右子树
            res=pNode.right;
            while(res.left!=null)
                res=res.left;
        }else{  //没有右子树
            TreeLinkNode parent=pNode.next; //父节点
            if(parent!=null && pNode==parent.left) //是父节点的左结点
                res=parent;
            else{ //是父节点的右结点
                while(parent!=null && parent.next!=null && parent!=parent.next.left)
                    parent=parent.next;
                res=parent==null? null : parent.next;
            } 
        }
        return res;
    }
}