剑指offer--二叉树的下一结点（中序遍历思想，枚举各种情况）

题目描述

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

解题思路：

一种最朴素的方法：由于TreeLinkNode中包含有next指针，指向当前结点的父节点，这样可以找到整棵二叉树的根节点。题目要求中序遍历的下一个节点，可以预处理获取整棵树的中序遍历序列，保存在一个容器中，之后for循环遍历容器中的节点，当访问到当前节点与给定节点相同时，直接return容器中的下一个节点即可。这种方法没有写。

以上是机试的思路，越简单越好，样例能够就好了。

但是如果试面试的过程中，用这种方法，就不是面试官想要的结果了。

接下来，分析一下中序遍历的特点以及二叉树中可能出现的节点的情况。对于二叉树的中序遍历而言，A：如果当前结点有右子树，那么当前结点作为中间结点，中序遍历为左子树，根结点，右子树。那么右子树的第一个访问的结点又是最左边的结点。所以这种情况，直接返回右子树的最左边的结点即可。B：当前结点没有右子树，那么当从左子树遍历完，访问根节点之后，下一个访问的点所在的层次肯定在当前结点的层数之上，因此需要向上"回溯"，找到把当前结点所在子树当作左子树的那个结点返回即可。C：当前结点作为结束结点，即中序遍历的最后一个结点，不会存在下一个结点，直接返回null即可。代码如下：

 

/*
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)
    {
        if(pNode == null) return pNode ;
        
        //当前节点有右节点
        while(pNode.right != null) {
            pNode = pNode.right ;
            while(pNode.left != null) {
                pNode = pNode.left ;
            }
            return pNode ;
        }
        
        //当前节点作为父节点的左节点或右节点(没有右节点代表，当前节点下一个节点的深度只会在当前层上面)
        while(pNode.next != null) {
            TreeLinkNode temp = pNode.next ;
            if(temp.left == pNode) {
                return temp ;
            }
            pNode = pNode.next ;
        }
        return null ;
    }
}