[JS] 프로그래머스 - 가장 먼 노드

문제 설명

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

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

 

제한 사항

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

입출력 예

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

 

입출력 예 설명

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

 

[그림 1] 예시

 

나의 풀이

 이 문제의 유형은 그래프를 이용한 문제 풀이 해결이다. 따라서 BFS를 풀어야한다. DFS는 시간초과를 초래할 것이다. 

따라서 나는 BFS를 근본으로 두어 노드 1로부터의 거리정보를 담은 dist와 각 노드 방문여부를 보여주는 Visited 배열을 만들었다.

 

전체적인 흐름은 다음과 같다.

1. 2차원 배열 edge를 연결리스트 형태의 그래프를 만든다(2차원 배열).

2. q, visited, dist를 이용하여 BFS 탐색을 한다.

3. BFS탐색 완료한 후에 모든 거리 정보가 담겨있는 dist 원소 중 최대 값을 구한다.

4. dist원소 중 최대값이 같은 원소들을 추출한다.

5. 추출한 원소의 개수를 반환한다.

 

const solution = (n, edge) => {
  const graph = getGraph(n, edge);
  const visited = new Array(n).fill(false);
  const dist = new Array(n).fill(0);
  const q = [];

  q.push({v:0, curDist:0});
  visited[0] = true;

  BFS(dist, q, visited, graph);

  const max = Math.max(...dist);
  return dist.filter((v) => v === max).length;
}

const BFS = (dist, q, visited, graph) => {
  while(q.length){
    const {v, curDist} = q.shift();
    const adj = graph[v];

    for(let i=0; i<adj.length; i++){
      const v = adj[i]-1;
      if(!visited[v]){
        visited[v] = true;
        q.push({v, curDist: curDist+1});
      }
    }
    dist[v] = curDist;
  }
}

const getGraph = (n, edge) => {
  const graph = Array.from({length:n}, (_) => []);
  while(edge.length){
    const [from, to] = edge.shift();
    graph[from-1].push(to);
    graph[to-1].push(from);
  }
  return graph;
}