二叉树的下一个节点

来源:互联网 发布:python黑帽子中文 编辑:程序博客网 时间:2024/05/17 16:54

给定二叉树和其中一个节点,如何找出中序遍历序列的下一个节点?树中的节点有两个指向左、右子节点的指针,还有一个指向父节点的指针。

分为三种情况:

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