剑指offer-- 二叉树的下一个结点

题目描述

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

解题思路

1. 这个题其实求的是中序线索二叉树某一结点的下一个结点
2. 如果一个结点有右子树，那么中序遍历中该结点的下一个结点为右子树中最左边的结点
3. 如果一个结点没有右子树，且它的父节点的左子树是该结点，那么父节点是该结点在中序遍历中的下一个结点
4. 如果一个结点没有右子树，且它的父节点的左子树不是该结点，则一直向上寻找父节点，直到找到一个是它父节点的左子节点的结点，该结点即为所求。
5. 其他情况为 null。

AC代码

/*
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;
        TreeLinkNode tmp = null;
        //有右子树
        if(pNode.right != null){
            tmp = pNode.right;
            while(tmp.left != null) tmp = tmp.left;
            return tmp;
        }
        //有父节点
        if(pNode.next != null){
            //父节点的左节点是该节点
            tmp = pNode.next;
            if(tmp.left == pNode) return tmp;
            while(tmp != null && tmp.left != pNode){
                pNode = tmp;
                tmp = tmp.next;
            }
            return tmp;
        }
            return null;
    }
}