B4016,求助!代码部分RE了,怀疑是否是Python递归过深
  • 板块学术版
  • 楼主zlj18279118766
  • 当前回复3
  • 已保存回复3
  • 发布时间2024/11/7 15:52
  • 上次更新2024/11/7 19:23:09
查看原帖
B4016,求助!代码部分RE了,怀疑是否是Python递归过深
1534447
zlj18279118766楼主2024/11/7 15:52
from collections import defaultdict
n=int(input())
edges=defaultdict(list)
for i in range(n-1):
    u,v=map(int,input().split())
    edges[u].append(v)
    edges[v].append(u)
res=[0]
d=[0]*(n+1)
def dfs(x,fa):
    for e in edges[x]:
        if e==fa:
            continue
        dfs(e,x)
        res[0]=max(res[0],d[x]+d[e]+1)
        d[x]=max(d[x],d[e]+1)
dfs(1,-1)
print(res[0])
2024/11/7 15:52
加载中...