题目描述
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
/*
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;
TreeLinkNode node=null;
//该结点有右子树,下一个结点就是该结点右子树最左结点
if(pNode.right!=null){
node=pNode.right;
while(node.left!=null)
node=node.left;
return node;
}
//该结点无右子树,分两种情况
else {
if(pNode.next==null)
return null;
//1 该结点是父节点的左孩子,则下个结点即为该结点的父节点
if(pNode==pNode.next.left){
return pNode.next;
}else{
//2 该结点是父节点的右孩子,则下个结点即为该结点的一个父节点的左结点
node=pNode.next;
while(node.next!=null){
if(node==node.next.left){
return node.next;
}
node=node.next;
}
}
}
return null;
}
}