二叉数的下一个节点

package com.niuke;

/**
 * Created by admin on 2018/3/11.
 * 给定一个二叉树和其中的一个结点，请找出中序遍历顺序的下一个结点并且返回。
 * 注意，树中的结点不仅包含左右子结点，同时包含指向父结点的指针。
 */
public class GetNext {
  //1 若该节点存在右子树：则下一个节点为右子树最左子节点
  //2 若该节点不存在右子树：这时分两种情况：
  //(1) 该节点为父节点的左子节点，则下一个节点为其父节点
  //(2) 该节点为父节点的右子节点，则沿着父节点向上遍历，直到找到一个节点的父节点的
  //    左子节点为该节点，则该节点的父节点下一个节点
    public TreeLinkNode getNext(TreeLinkNode pNode) {
        if(pNode==null) {
            return null;
        }
        if(pNode.right!=null) {//该节点存在右子树
            pNode=pNode.right;
            while (pNode.left!=null) {//找到最左子节点
                pNode=pNode.left;
            }
            return pNode;
        } else if(pNode.next!=null&&pNode.next.left==pNode) {
            //该节点不存在右子树,且为父节点的左子节点
            return pNode.next;
        } else if(pNode.next!=null&&pNode.next.right==pNode) {
            //该节点不存在右子树,该节点为父节点的右子节点
            while (pNode.next!=null&&pNode!=pNode.next.left) {
                pNode=pNode.next;
            }
            return pNode.next;
        } else {//该节点不存在右子树,该节点无父节点，为根节点
            return null;
        }
    }
}

class TreeLinkNode {
    int val;
    TreeLinkNode left = null;
    TreeLinkNode right = null;
    TreeLinkNode next = null;//指向父节点的指针

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