美文网首页数据结构与算法
二叉树的下一个节点

二叉树的下一个节点

作者: 而立之年的技术控 | 来源:发表于2019-12-15 19:02 被阅读0次
WechatIMG1.jpeg
def get_next(p_node):
    """
    获取二叉树(p_node)的下一个节点
    """
    # 针对的是 第1中情况
    if p_node.right:
        tmp = p_node.right
        while tmp.left:
            tmp = tmp.left
        return tmp
    # 针对 2、3、4情况    
    else:
        tmp = p_node
        while tmp.next:
            if tmp.next.left == tmp:
                return tmp.next
            tmp = tmp.next
    return None

相关文章

网友评论

    本文标题:二叉树的下一个节点

    本文链接:https://www.haomeiwen.com/subject/bpllnctx.html