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

yes3427·2021년 9월 7일
0

문제풀이

목록 보기
4/9
post-thumbnail

https://programmers.co.kr/learn/courses/30/lessons/49189

문제 설명

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

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

제한사항

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

입출력 예

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;
}
profile
소비자가 아닌 생산자가 되자

0개의 댓글