剑指offer系列31-----二叉树的下一个节点

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

package com.exe7.offer;

/**
 * 【题目】给定一个二叉树和其中的一个结点，请找出中序遍历顺序的下一个结点并且返回。
 *            注意，树中的结点不仅包含左右子结点，同时包含指向父结点的指针。
 * @author WGS
 *
 */
public class GetNextNode {

    public class TreeNode{
        int val;
        TreeNode left=null;
        TreeNode right=null;
        TreeNode next=null;
        public TreeNode(int val){
            this.val=val;
        }
    }
    
    
    public TreeNode getNextNode(TreeNode pNode){
        if(pNode==null) return null;
        //1 判断有无右子树，有就是右子树最左子节点；
        TreeNode curNode=null;
        if(pNode.right!=null){
            curNode=pNode.right;
            while(curNode.left!=null){
                curNode=curNode.left;
            }
            return curNode;
        }
        //2 无右子树;
        else{
            //1) 判断该结点是否是其父节点的左结点，是，下个结点就是其父节点。
            if(pNode.next==null) return null;
            if(pNode==pNode.next.left){
                return pNode.next;//返回其父节点
            }
            
            
            //2) 该结点不是父节点的左结点即是其右节点，则判断其父节点是否是父父结点的左结点；
            curNode=pNode.next;//该结点的父节点
            while(curNode.next!=null){
                if(curNode==curNode.next.left){
                    return curNode.next;//返回其父节点的父节点
                }
                curNode=curNode.next;//继续回溯
            }
        }    
        
        return null;
        
    }
}