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

프로그래머스 [Lv. 2] 전력망을 둘로 나누기 {언어 : JavaScript} 본문

알고리즘

프로그래머스 [Lv. 2] 전력망을 둘로 나누기 {언어 : JavaScript}

스위태니 2024. 3. 17. 01:40

문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/86971

 

프로그래머스

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

programmers.co.kr

정답 코드

function solution(n, wires) {
    let answer = 101;
    const adjL = Array(n+1).fill().map(() => [])
    wires.forEach(([s, e]) => {
        adjL[s].push(e);
        adjL[e].push(s);
    })
    const bfs = (s, e) => {
        let V = Array(n+1).fill().map((e) => 0);
        let cnt = 1;
        const q = [1];
        V[1] = 1;
        while (q.length > 0) {
            const start = q.shift();
            for (const next of adjL[start]) {
                if (V[next] !== 0 || (s === start && e === next) || (e === start && s === next)) {
                    continue
                }
                V[next] = 1;
                cnt += 1;
                q.push(next)
            }
        }
        return cnt * 2;
    }
    
    wires.forEach(([s, e]) => {
        answer = Math.min(Math.abs(n - bfs(s, e)), answer)
    })
    return answer;
}

풀이 방법

  1. adjL을 구성한다.
  2. bfs를 만들고 1번 부터 순회할 수 있도록 한다.
    1. 현재 s, e 또는 e, s가 자르려고 하는 전선이므로 start와 next가 일치하는지 비교한다.
      1. 이 과정에서 파이썬은 (e, s) == (start, next)와 같이 비교하지만 자바스크립트는 그럴 수 없었다.
    2. cnt * 2를 return한다.
      1. 현재 이동한 송전탑의 개수 = cnt
      2. 이동하지 않은 송전탑의 개수 = n - cnt
      3. 전력망을 둘 로 나눈 후 송전탑의 차이 = | n - cnt * 2 |
  3. answer와 비교해서 최소값을 출력한다.

느낀점

 

자바스크립트 [배열 비교]

문제 발견 파이썬의 경우 배열을 직접 비교하는 것이 가능하다. 자바스크립트는 직접 비교할 경우 빈 배열이라고 할지라도 false를 반환한다. 파이썬 코드 A = [1, 2, 3] B = [1, 2, 3] C = [1, 2] print(A == B,

sound-programming.tistory.com

 

자바스크립트 [빈 배열을 요소로 가진 배열 만들기]

코드 // 자바스크립트로 파이썬 리스트 컴프리헨션과 유사한 기능 만들기 // arr = [[] for _ in range(n)] const n = 5; // 원하는 길이 const arr1 = Array.from({ length: n }, () => []); const arr2 = [...Array(n)].map(() => []); c

sound-programming.tistory.com