剑指offer-二叉树中的下一个节点(Java)

问题描述

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

解析

要找到中序遍历下的下一个节点。这个节点可以分为两种情况
一、该节点有右子树
二、该节点没有右子树
第一种比较处理起来比较简单，直接将其右节点进行中序遍历即可，并将一个遍历到的最右节点返回。
第二种情况又分为两种情况
1、该节点是父节点的左子节点
2、该节点是父节点的右子节点
这里，该节点是父节点的左子节点的这种情况比较简单，直接将父节点返回即可。
如果是父节点的右子节点的话，需要不断的向上移动，直到对应的节点不是父节点的右节点（即左节点），，既然他是父节点的左节点，此时将这个节点父节点返回即可，或者遍历到了根节点，返回null。

代码

public 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;
}