题目:给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
题解:如果一个结点有右子树,那么它的下一个结点是它右子树中的最左子节点;
如果一个结点没有右子树,并且它是父节点的左节点,则它的下一个结点是父节点;
如果一个结点没有右子树,并且它是父节点的右结点,则沿着指向父节点的指针一直向上遍历,直到找到一个是它父节点的左子节点的结点。
/*
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 pNext = null;
if(pNode.right!=null){
TreeLinkNode right = pNode.right;
while(right.left!=null){
right = right.left;
}
pNext = right;
}else if (pNode.next!=null) {
TreeLinkNode cur = pNode;
TreeLinkNode pa = cur.next;
while(pa!=null&&cur==pa.right){
cur = pa;
pa = pa.next;
}
pNext = pa;
}
return pNext;
}
}