二叉树的下一个节点

题目描述

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

 

/*
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 first(TreeLinkNode p) {
        if (p.left != null) {
            return first(p.left);
        }
        return p;
    } 
    public TreeLinkNode GetNext(TreeLinkNode pNode)
    {
        TreeLinkNode p = pNode;
        TreeLinkNode last = null;
        while (p != null) {
            if (p.left != null && p.left == last) {
                return p;
            }
            if (p.right != last && p.right != null) {
                return first(p.right);
            }
            last = p;
            p = p.next;
        }
        return p;
        
    }
}