[백준][Python] 1240번 노드사이의 거리

1240번: 노드사이의 거리

 

 

코드

import sys
from collections import defaultdict

def find(a,b):
	queue = []
	queue.append((a,0))
	visit[a] = True
	while queue:
		next,len = queue.pop()
		if next == b:
			return len
		for i,l in dic[next]:
			if visit[i]:
				continue
			visit[i] = True
			queue.append((i,l+len))

n,m = map(int,input().split())
dic = defaultdict(list)

for i in range(n-1):
	a,b,l = map(int,sys.stdin.readline().split())
	dic[a].append((b,l))
	dic[b].append((a,l))
for i in range(m):
	visit = [False]*(n+1)
	a,b = map(int,sys.stdin.readline().split())
	print(find(a,b))