美团笔试-图的最短路径-树的深度-python

最新推荐文章于 2024-04-30 21:57:26 发布

原创

最新推荐文章于 2024-04-30 21:57:26 发布 · 1.2k 阅读

1 ·

CC 4.0 BY-SA版权

这道笔试题要求求解图的最短路径，但不必保存具体路径。通过理解题意，可以避免枚举所有可能，采用先遍历短路径子树再遍历长路径子树的策略。提供了两种解法，分别是利用Python数组列表保存元数据和C++的深度优先搜索（DFS）求最大深度。

这道题不要求保存路径，因此就不用列举所有的可能性（实习也很难遍历完每种可能），上图题解就是一个很好的思考过程，我们不需要一个一个尝试，只需知道有种路径一定是先遍历短路径子树，最后再遍历长路径子树就可以了。

输入输出类似：因为实在找不到原题了

第一种解法：利用python数组列表保存元数据


def min_path(n, B):
    node_list = [0] * (n + 1)  # 保存节点的距离
    for path in B:
        # 遍历路径，使后一节点距离值等于前一节点距离值加一
        x, y = path
        node_list[y] = node_list[x] + 1
    depth = 0
    # 找到最大深度
    for i in range(1, n + 1):
        if node_list[i] > depth:
            depth = node_list[i]
    # 计算最短路径,即最深的路径走一遍，其他路径走2遍
    return 2 * n - 2 - depth
 
 
if __name__ == '__mai