몸과 마음이 건전한 SW 개발자

프로그래머스 [Lv. 3] 가장 먼 노드 {언어 : JavaScript} 본문

알고리즘

프로그래머스 [Lv. 3] 가장 먼 노드 {언어 : JavaScript}

스위태니 2024. 6. 19. 19:42

문제 링크

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

정답 코드

function solution(n, edge) {
    let answer = 0;
    let maxD = 0;

    const queue = [];
    
    const adjL = Array(n+1).fill().map(() => []);
    for (const [start, to] of edge) {
        adjL[start].push(to);
        adjL[to].push(start);
    }
    let visited = Array(n+1).fill().map(() => -1);
    
    q = [1];
    visited[1] = 0;
    while (q.length) {
        const s = q.shift();
        for (const nd of adjL[s]) {
            const dist = visited[s] + 1;
            let isValid = false;
            if (visited[nd] == -1) {
                isValid = true;
            } else {
                if (dist < visited[nd]) {
                    isValid = true;
                };
            };
            if (isValid) {
                visited[nd] = dist;
                q.push(nd);
                if (maxD < dist) {
                    maxD = dist;
                    answer = 1;
                } else if (maxD == dist) {
                    answer += 1;
                };
            };
        };
    };
    return answer;
}

풀이 방법

  1. 노드 배열을 만든다.
  2. visited라는 방문 배열을 만든다.
    1. 이거를 dp로 사용한다.
  3. 현재 도달한 지점의 값이 -1이면 새로운 거리를 치환한다.
  4. 이미 지나갔으면 현재 지나온 거리에 1을 더한 거리보다 큰지 비교한다.
  5. 크다면 새로운 거리로 치환한다.
  6. 거리가 maxD보다 크다면 maxD는 dist가 되고 answer = 1이 된다.
  7. maxD와 같다면 answer에 1을 더해준다.
  8. answer를 반환하면 끝!

느낀점

  • 코드를 더 가독성 높게 만들고 싶고 효율적으로 만들고 싶다.