寻找中序遍历下一个节点

题目描述

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

思路

可以发现，情况分为两类：
- 第一类：给定的节点包含右子树，则下一个节点一定在右子树中。是这个节点右子树的最左子节点。
- 第二类：给定的节点不包含右子树。有两种情况：
- 该节点是父节点的左子树，则下一个节点就是这个父节点。
- 该节点是父节点的右子树，则寻找该节点的父节点的父节点….直到当前结点是其父节点的左孩子位置。如果一直到其父节点是null（根节点），则返回null。

code

/*
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.right!=null){
            pNode=pNode.right;
            while(true){
                if(pNode.left==null){
                    return pNode;
                }else{
                    pNode=pNode.left;
                }
            }
        }else{
            while(true){
                if(pNode.next==null){
                    return null;
                }
                if(pNode.next.left==pNode){
                    return pNode.next;
                }else{
                    pNode=pNode.next;
                }
            }
        }

    }
}