프로그램/Python
[프로그래머스 문제풀이] 가장 먼 노드
naudhizb
2022. 4. 19. 23:39
반응형
그래프를 탐색하며 가장 먼 노드를 찾는 문제이다.
나는 BFS를 이용하여 거리를 구한 뒤 가장 먼노드를 찾았다.
def solution(n, edge):
answer = 0
v = [set() for _ in range(n+1)]
visit = [-1] * (n+1)
for e in edge:
v[e[0]].add(e[1])
v[e[1]].add(e[0])
def bfs(root):
from collections import deque
queue = deque()
queue.append((root, 0))
visit[root] = 0
while queue:
n, l = queue.popleft()
for e in v[n]:
if visit[e] == -1:
visit[e] = l+1
queue.append((e, l+1))
bfs(1)
#print(v)
#print(visit)
answer = visit.count(max(visit))
return answer
```
반응형