给定二叉树和其中一个节点,如何找出中序遍历序列的下一个节点?树中的节点有两个指向左、右子节点的指针,还有一个指向父节点的指针。
分为三种情况:
1. 如果一个节点有右子树,那么它的下一个节点就是它的右节点的最左子节点。
2. 如果一个节点没有右子树,并且该节点是它父节点的左子节点,那么它的下一节点就是它的父节点。
3. 如果一个节点没有右子树,并且还是它父节点的右子节点,那么从该节点开始一直沿着父节点的指针向上遍历,直到找到一个节点是它父节点的左子节点,那么该节点的下一节点就是这个节点的父节点。
2与3都是在该节点没有右子树的情况下,都是要找出该节点是父节点的左子节点的父节点。如果该节点恰好是父节点的左子节点,那么父节点即为下一个节点。
如果该节点不是父节点的左子节点,那么一直向上遍历,直到找到一个节点为父节点的左子节点,那么该父节点即为下一节点。
2与3的情况下一节点都是根节点。
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&&pNode.next.right==pNode) {
pNode=pNode.next;
}
return pNode.next;
}