二叉树的下一个结点

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

/**
 * 二叉树的下一个结点
 * 
 * @author 过路的守望
 *
 */
public class GetNextNode {

    public TreeLinkNode getNext(TreeLinkNode node) {
        /*
         * 当前节点为空，返回null
         */
        if (node == null) {
            return null;
        }
        /*
         * 若当前节点有右儿子，则返回右儿子的最左儿子
         */
        if (node.right != null) {
            node = node.right;
            while (node.left != null) {
                node = node.left;
            }
            return node;
        }
        /*
         * 若当前节点无右儿子，则找第一个当前节点是父节点左孩子的节点
         */
        while (node.next != null) {
            if (node.next.left == node) {
                return node.next;
            }
            node = node.next;
        }
        return null;
    }

}

class TreeLinkNode {
    int val;
    TreeLinkNode left = null;
    TreeLinkNode right = null;
    TreeLinkNode next = null;

    TreeLinkNode(int val) {
        this.val = val;
    }
}