剑指offer-57-二叉树的下一个结点 -- Java实现

题目

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

分析

思路一：

分三种情况：
1、 当前节点有右子树，则下一节点为右子树中的最左子节点，如果右子树无左子节点则为右子节点本身。
2、 如果没有右子树，如果父节点的左子节点就是该节点本身，则下一节点即父节点。
3、 如果既没有右子树，又不是父节点的左子节点，如下图的J节点，则其应往上溯，直到存在一个向上遍历的过程中存在一个节点的父节点的左子节点正好是该节点自身，则这个父节点就是下一个节点。如下图J的父节点为E，它的父节点左子节点是D而不是它，继续向上，B的左子节点是D而非E，所以继续向上，A的左子节点就是B，所以J的下一节点就是A。

时间复杂度：o(N)
空间复杂度：o(N)

代码：

/*
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 null;
        //1. 如果该节点有右子树，则返回右子树的最左子节点
        if(pNode.right!=null) {
            pNode = pNode.right;
            while(pNode.left!=null) pNode = pNode.left;
            return pNode;
        }
        //2. 如果该节点无右子树，返回一个父节点是它左子节点的节点
        while(pNode.next!=null) {
            if(pNode.next.left==pNode) return pNode.next;
            pNode = pNode.next;
        }
        //3. 遍历到根节点仍没找到则为null
        return null;
    }
}