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、如果该节点没有右子树,该节点为父节点的右子树,向上找,一直到一个节点是其父节点的左子节点,下一个节点为该父节点;如果找不到,则没有下一个节点。

posted @ 2019-08-12 20:33  尘世中一个迷途小书童  阅读(116)  评论(0编辑  收藏  举报