https://programmers.co.kr/learn/courses/30/lessons/49189
n개의 노드가 있는 그래프가 있습니다. 각 노드는 1부터 n까지 번호가 적혀있습니다. 1번 노드에서 가장 멀리 떨어진 노드의 갯수를 구하려고 합니다. 가장 멀리 떨어진 노드란 최단경로로 이동했을 때 간선의 개수가 가장 많은 노드들을 의미합니다.
노드의 개수 n, 간선에 대한 정보가 담긴 2차원 배열 vertex가 매개변수로 주어질 때, 1번 노드로부터 가장 멀리 떨어진 노드가 몇 개인지를 return 하도록 solution 함수를 작성해주세요.
n vertex return
6 [[3, 6], [4, 3], [3, 2], [1, 3], [1, 2], [2, 4], [5, 2]] 3
입출력 예 설명
예제의 그래프를 표현하면 아래 그림과 같고, 1번 노드에서 가장 멀리 떨어진 노드는 4,5,6번 노드입니다.
n이 큰 경우에는 엔진이 최적화를 해주지 않기 때문에
연결리스트를 사용한다.
shift와 unshift 사용은 피해야 한다.
우선 연결리스트를 구현한다.
enqueue
, dequeue
, isEmpty
class Queue {
constructor() {
this.queue = [];
this.front = 0;
this.rear = 0;
}
enqueue(value) {
this.queue[this.rear++] = value;
}
dequeue() {
const value = this.queue[this.front];
delete this.queue[this.front];
this.front += 1;
return value;
}
isEmpty() {
return this.front === this.rear;
}
}
핵심 코드이다.
BFS를 사용하였다.
function solution(n, edge) {
// 얕은 복사
/*
[[], [], []... ]
*/
const graph = Array.from(Array(n + 1), () => []);
// 양방향구조
for (const [src, dest] of edge) {
graph[src].push(dest);
graph[dest].push(src);
}
// distance = [0,0,0,...]
const distance = Array(n + 1).fill(0);
distance[1] = 1;
//BFS
const queue = new Queue();
queue.enqueue(1);
while (!queue.isEmpty()) {
const src = queue.dequeue(); //shift는 O(n)이지만 요소가 적을 경우에는 자바스크립트 엔진에서 최적화를 해준다.
for (const dest of graph[src]) {
if (distance[dest] === 0) {
queue.enqueue(dest);
distance[dest] = distance[src] + 1;
}
}
}
const max = Math.max(...distance);
return distance.filter((item) => item === max).length;
}