[programmers] 가장 먼 노드

코딩 테스트 연습 - 그래프

 

---

<가장 먼 노드>

문제 설명

 

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

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

 

 

제한 사항

 

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

 

 

문제 풀이

  - my solution

def bfs(graph, start, depth):
    queue=[start]

    while queue:
        temp=queue.pop(0)
        for i in graph[temp]:
            if depth[i]==0 and i!=1: # 아직 방문하지 않은 곳
                depth[i]=depth[temp]+1 #depth
                queue.append(i)         
    return depth
    
def solution(n, edge):
    answer = 0
    
    graph={}
    depth=[0]*(n+1)
    
    for i in range(n):
        graph[i+1]=[]
    
    for i in edge: #graph
        graph[i[0]].append(i[1])
        graph[i[1]].append(i[0])
        

    result=bfs(graph,1, depth)
    
    find=max(result) #가장 멀리 떨어진 노드
    
    for value in result:
        if value==find:
            answer+=1
    
    return answer

 

처음에는 시간 초과가 발생하였다. 

 

❓ 왜 시간 초과가 발생했는가?

더 이상 시간을 줄일 방법이 생각나지 않아 검색의 힘을 빌렸다. 이것 때문인지는 모르겠지만 내가 생각한 것으로는 방문을 확인하는 list와 거리를 저장하는 dict를 따로 구현했기 때문이다. 이것을 list 하나로 합쳐 구현했더니 시간 초과 문제를 해결할 수 있었다. 

 

 

• bfs
  - queue: [노드]
  - queue가 빌 때까지 반복 -> queue의 맨 앞 값 뽑기 -> 그 노드(x)와 연결된 노드(y) 탐색 -> 연결된 노드(y)의 depth 값이 0이고 1이 아닌 다른 노드이면 아직 방문하지 않은 것 -> 연결된 노드(y)의 depth 값: 노드(x)의 depth +1 -> queue에 연결된 노드(y) 추가
• main
  - graph: dict, 양방향 추가
  - depth: list, n+1개 만들어서 거리 표시
  - result: bfs 호출하여 얻은 depth 결과
  - find: 1번 노드로부터 가장 멀리 떨어진 노드
  - 1번 노드로부터 가장 멀리 떨어진 노드를 찾기 위해 반복문

---

생각🤔

 

처음에 구현한 후 시간 초과가 발생했을 때 정말 막막했었다. 

생각보다 쉽게 구현했으며 더 이상 시간을 줄일 곳이 없다고 생각했기 때문이었다. 하지만 검색을 통해 내가 생각보다 많은 list, dict를 통해 불필요한 저장을 하고 있다는 것을 알게 되었으며 이것을 해결하였더니 문제를 통과할 수 있었다.

---

 

출처: 프로그래머스 코딩 테스트 연습, https://programmers.co.kr/learn/challenges