剑指Offer——P8(二叉树的下一个节点)

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

/*
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如果该节点的右子树为空
		/*
		 * 1)如果右子树为空，并且该节点是其父节点的左子树，则下一个节点为其父节点
		 * 2）如果右子树为空，并且该节点是其父节点的右子树，则遍历其父节点，直到当前节点是其父节点的左孩子，则下一个节点为其父节点
		 */
		while(pNode.next != null) {
			if(pNode.next.left == pNode) return pNode.next;
			pNode = pNode.next;
		}
		return null;
    }
}