[python] 프로그래머스 Level3 가장 먼 노드 / 그래프

https://programmers.co.kr/learn/courses/30/lessons/49189?language=python3

문제 설명

n개의 노드가 있는 그래프가 있습니다. 각 노드는 1부터 n까지 번호가 적혀있습니다. 1번 노드에서 가장 멀리 떨어진 노드의 갯수를 구하려고 합니다. 가장 멀리 떨어진 노드란 최단경로로 이동했을 때 간선의 개수가 가장 많은 노드들을 의미합니다.

노드의 개수 n, 간선에 대한 정보가 담긴 2차원 배열 vertex가 매개변수로 주어질 때, 1번 노드로부터 가장 멀리 떨어진 노드가 몇 개인지를 return 하도록 solution 함수를 작성해주세요.

제한사항

• 노드의 개수 n은 2 이상 20,000 이하입니다.
• 간선은 양방향이며 총 1개 이상 50,000개 이하의 간선이 있습니다.
• vertex 배열 각 행 [a, b]는 a번 노드와 b번 노드 사이에 간선이 있다는 의미입니다.

입출력 예

nvertexreturn

n	vertex	return
6	[[3, 6], [4, 3], [3, 2], [1, 3], [1, 2], [2, 4], [5, 2]]	3

입출력 예 설명

예제의 그래프를 표현하면 아래 그림과 같고, 1번 노드에서 가장 멀리 떨어진 노드는 4,5,6번 노드입니다.

문제 풀이

bfs를 사용해서 풀었다.

1. 딕셔너리에 노드별 연결 정보를 넣는다.

2. 효율성을 위해 deque를 사용한다.(일반 리스트에 pop(0)을 하면 매번 O(n)의 비효율적 복잡도)

3. deque에 [노드 번호, 깊이] 정보를 한번에 삽입한다.

4. check 배열은 -1로 초기화하여 준비하고, 인접 노드를 탐색할 땐 한번 deque에 넣은 노드는 0으로 check한다. (0으로 체크를 안해주면 이미 탐색한 노드를 재탐색함)

5. 큐에서 꺼내올 때, check[노드 번호] = 깊이 로 넣어준다.

6. 마지막으로 check배열에서 최댓값을 카운트해주면 된다.

solution.py

나의 풀이

from collections import deque

def solution(n, edge):
    routes = dict()
    for e1, e2 in edge:
        routes.setdefault(e1, []).append(e2)
        routes.setdefault(e2, []).append(e1)
        
    q = deque([[1, 0]]) # node number, depth
    check = [-1]*(n+1)
    while q:
        index, depth = q.popleft()
        check[index] = depth
        for i in routes[index]:
            if check[i]==-1:
                check[i] = 0
                q.append([i, depth+1])
        depth += 1
    return check.count(max(check))