剑指Offer37-二叉树的下一个结点
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
/* 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;
//如果当前结点有右子树,那么寻找右子树中的最左结点
if(pNode.right!=null){
pNode=pNode.right;
while(pNode.left!=null){
pNode=pNode.left;
}
return pNode;
}
//如果当前结点没有右子树,那么返回第一个结点的父结点的左结点正好是当前结点的 父结点
while(pNode.next!=null){
if(pNode.next.left==pNode)
return pNode.next;
pNode=pNode.next;
}
return null;
}
}