二叉数的下一个节点

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;
}
}