Brise

[프로그래머스 문제풀이] 가장 먼 노드 본문

프로그램/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
    ```
반응형
Comments