1240번/DFS 노드사이거리/파이썬

📕 문제

https://www.acmicpc.net/problem/1240

 

📗실행결과

 

 

📄  풀이

리스트 생성 비교

https://m.blog.naver.com/wjddudwo209/221253421426

python [[]] vs [[] for _ in range(n)]

 

💻 내 코드

 

import sys
input = sys.stdin.readline
sys.setrecursionlimit(1000000)

n,m = map(int, input().split())

graph = [[] for i in range(n+1)]

for i in range(n-1):
    a,b,c = map (int, input().split())
    graph[a].append((b,c))
    graph[b].append((a,c))

# 인접노 확인하기 
def dfs(x):
    # x의 인접 노드들 하나씩 확인
    for data in graph[x]:
        y = data[0]
        # x에서 y로 가는 간선 비용
        cost = data[1]

        # 방문하지 않은 노드라면
        if not visited[y]:
            # 방문처리
            visited[y] = True
            distance[y] = distance[x] + cost
            dfs(y)

for i in range(m):
    x, y = map(int, input().split())

    # 방문 여부 및 최단 거리 테이블 초기화
    visited = [False] * (n+1)
    distance = [-1] * (n+1)
    
    visited[x] = True
    distance[x] = 0
    dfs(x)
    print(distance[y])

 

💻 답안

  //

 

☺️ 내 코드에서 보완해야 할 점

 

 

😉 참고

 

 

 

https://blog.naver.com/wjavmtngkr1/223202285334

백준 1240 노드사이의 거리 ( dfs 문제 )