美团2019秋招笔试—图的遍历

最新推荐文章于 2025-08-10 10:46:21 发布

原创最新推荐文章于 2025-08-10 10:46:21 发布 · 403 阅读

0 ·

CC 4.0 BY-SA版权

笔试记录专栏收录该内容

8 篇文章

订阅专栏

本文探讨了在N个节点、N-1个边的无向连通图中寻找最短路径的方法，通过生成树的概念，阐述了如何确定最大深度，并基于此计算最短总路程，提供了一段Python代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：

思路：N个节点，N -1个边，且无向连通图，每次从1开始遍历，说明是生成树；则最长的边遍历一次，其他的边遍历两次，则能使得路程最短；首先求得最大深度m，然后就可以得到路程m + (N-1-m)*2;

代码：

n =7
edge = [[1,2],[2,3],[2,4],[3,6],[6,7],[3,5]]

from collections import defaultdict

d = defaultdict(list)
for i in edge:
    d[i[0]].append(i[1])
    d[i[1]].append(i[0])

visited = [False] * (n + 1)
deep = [0] * (n + 1)
parent = [-1] * (n + 1)

def recursion(d, node, deep_tmp):
    deep_tmp += 1
    visited[node] = True
    if d[node] == []:
        return 
    for i in d[node]:
        if visited[i] == False:
            deep[i] = deep_tmp
            parent[i] = node
            recursion(d, i, deep_tmp)
recursion(d, 1, 0)
max_deep = max(deep)
print(max_deep + 2 * (n - 1 - max_deep))