剑指offer编程题找出二叉树中序遍历的下一个结点Java实现（二叉树，中序遍历）

题目描述

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

问题分析

二叉树的结构如下所示，题目中的描述next意思是指向当前节点的父节点。我们要知道中序遍历的含义，先按中序遍历左子树，再输出当前结点，然后再中序遍历右子树。

public class TreeLinkNode {
    int val;
    TreeLinkNode left = null;
    TreeLinkNode right = null;
    TreeLinkNode next = null;

    TreeLinkNode(int val) {
        this.val = val;
    }
}

代码及讲解

按照中序遍历的特点，如果当前结点没有右子树，且它是它父结点的左子结点（例如D），则当前结点的下一个遍历的结点就是它的父结点（B），如果它是它父结点的右子结点（G，I）则它应该向上寻找到一个结点是该结点父结点的左子结点（G对应null，I对应A），则下一个遍历的就是该结点的父节点。
如果当前结点有右子树（如E，B），则按照中序遍历的规则，应该对当前节点的右子结点进行中序遍历，则下一个遍历的点为当前节点右子树中最下面的左子节点（H）。
我这么说肯定看的很茫然，找了个图当例子讲一下。
下图的中序遍历顺序为{D,B,H,E,I,A,F,C,G}。配合上面的讲述应该比较清楚了吧。

public class Solution {
    public TreeLinkNode GetNext(TreeLinkNode pNode)
    {
        if(pNode==null){
            return pNode;
        }
        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;
            if(pNode.next.right==pNode) pNode=pNode.next;
        }
        return null;
    }
}

总结

我觉得本题要注意对问题的分析，我开始写是从是否是叶子结点来判断的，最后写的一塌糊涂，把自己也绕了进去，后来看了一些讨论和题解才解出来。其实二叉树中序遍历找下一个结点并不难，注意找到问题的切入点才是关键，老板们加油，我有个朋友说过种一棵树最好的时间是十年前然后就是。