[24484] 알고리즘 수업 - 깊이 우선 탐색 6

Junyoung Park·2022년 3월 2일
0

코딩테스트

목록 보기
155/631
post-thumbnail

1. 문제 설명

알고리즘 수업 - 깊이 우선 탐색 6

2. 문제 분석

이전 문제와 동일. 내림차순 정렬에 주의.

3. 나의 풀이

import sys
from collections import deque

n, m, r = map(int, sys.stdin.readline().rstrip().split())

nodes = [[] for _ in range(n+1)]
visited = [False for _ in range(n+1)]
nodes_depth = [0 for _ in range(n+1)]
nodes_cnt = [0 for _ in range(n+1)]

for _ in range(m):
    tail, head = map(int, sys.stdin.readline().rstrip().split())
    nodes[tail].append(head)
    nodes[head].append(tail)

for i in range(n+1):
    nodes[i].sort()
stack = deque()
stack.append([r, 0])
cnt = 1
while stack:
    cur_node, depth = stack.pop()
    if visited[cur_node]: continue
    visited[cur_node] = True
    nodes_depth[cur_node] = depth
    nodes_cnt[cur_node] = cnt
    cnt += 1
    for next_node in nodes[cur_node]:
        if not visited[next_node]:
            stack.append([next_node, depth+1])

total = 0
for i in range(1, n+1):
    if i == r or nodes_depth[i] == 0: continue
    else: total += nodes_depth[i]*nodes_cnt[i]

print(total)
profile
JUST DO IT

0개의 댓글