两段代码
功能都是收集数据
就是第一行输出n,下面n-1行每行输出u,v,d,分别代表结点u,d,ud之间距离d
我用了两种方法
n=int(input())
point=[[] for i in range(n+1)]#n+1个顶点(有多余)point[i]=代表[i号点连接的顶点]
distance=[[float('inf')]*(n+1) for i in range(n+1)]#distance[i][j]i到j的距离
for i in range(n-1):
tmp=list(map(int,input().split()))
point[tmp[0]].append(tmp[1])
point[tmp[1]].append(tmp[0])
distance[tmp[0]][tmp[1]]=distance[tmp[1]][tmp[0]]=tmp[2]
上面这个是第一种,用两个列表存数据,一个列表存连接关系,一个列表存距离
但是这种方法在做题的时候出现内存超限了,,为啥?
我改变了数据结构,尝试使用字典,下面这段代码就满分通过了,为啥?
n=int(input())
point=dict((i,{})for i in range(n+1))#point=[0:{},1:{}....] point[i][j]代表i到j的距离,同时能表示两者有连接
res=[0 for i in range(n+1)]
for i in range(n-1):
tmp=list(map(int,input().split()))
point[tmp[0]][tmp[1]]=tmp[2]
point[tmp[1]][tmp[0]]=tmp[2]
第一种方法不就多了一个容器么,怎么差别会这么大