08-二叉树的下一个节点
题目:给定一棵二叉树和其中其中的一个节点,如何找出中序遍历的下一个节点?树中的节点除了有两个分别指向左右子节点的指针,还有一个指向父节点的指针。
def treeof2_next_node(head,node): hnode = head p = node if not hnode: return None if p.right: pright = p.right while pright.left: pright = pright.left return pright elif p.father.left == p: return p.father elif p.father.right == p: fnode = p.father if fnode.father == None: return None while fnode.father: if fnode.father.left == fnode: return fnode.father fnode = fnode.father return None
注:
中序遍历下一个节点有三种情况
1、如果该节点有右子树,下一个节点为右子树的最左节点
2、如果该节点没有右子树,该节点为父节点的左子树,下一个节点为其父节点
3、如果该节点没有右子树,该节点为父节点的右子树,向上找,一直到一个节点是其父节点的左子节点,下一个节点为该父节点;如果找不到,则没有下一个节点。