[프로그래머스][LEVEL2] 전력망을 둘로 나누기

# 문제 원문

 

n개의 송전탑이 전선을 통해 하나의 트리 형태로 연결되어 있습니다. 당신은 이 전선들 중 하나를 끊어서 현재의 전력망 네트워크를 2개로 분할하려고 합니다. 이때, 두 전력망이 갖게 되는 송전탑의 개수를 최대한 비슷하게 맞추고자 합니다.

송전탑의 개수 n, 그리고 전선 정보 wires가 매개변수로 주어집니다. 전선들 중 하나를 끊어서 송전탑 개수가 가능한 비슷하도록 두 전력망으로 나누었을 때, 두 전력망이 가지고 있는 송전탑 개수의 차이(절대값)를 return 하도록 solution 함수를 완성해주세요.

---

제한사항

• n은 2 이상 100 이하인 자연수입니다.
• wires는 길이가 n-1인 정수형 2차원 배열입니다.
  • wires의 각 원소는 [v1, v2] 2개의 자연수로 이루어져 있으며, 이는 전력망의 v1번 송전탑과 v2번 송전탑이 전선으로 연결되어 있다는 것을 의미합니다.
  • 1 ≤ v1 < v2 ≤ n 입니다.
  • 전력망 네트워크가 하나의 트리 형태가 아닌 경우는 입력으로 주어지지 않습니다.

---

입출력 예

n / wires / result

9	[[1,3],[2,3],[3,4],[4,5],[4,6],[4,7],[7,8],[7,9]]	3
4	[[1,2],[2,3],[3,4]]	0
7	[[1,2],[2,7],[3,7],[3,4],[4,5],[6,7]]	1

 

# 문제 풀이

 

아직도 DFS, BFS는 어렵고 헷갈린다. 모든 와이어를 한번씩 끊어보고 두 그룹의 차이가 가장 적은 경우를 구해서 그 그룹사이의 송전탑 차이를 반환해주는 문제이다. 처음에는 그냥 반복문으로만 될거라 생각했는데 입력 받은 wires의 요소가 순서대로 주어진다는 보장이 없다는 점 때문에 DFS로 전환해서 송전탑들을 카운팅했다.그렇게 구해진 수중 가장 작은 수를 반환하면 된다.

 

# 솔루션 플로우

1. 입력 받은 wries의 길이 만큼 반복한다.

2. 입력 받은 wries의 가장 첫 요소를 배열에서 추출하고 (전선 끊기) 송전탑 갯수를 체크한다.

3. 체크 방식은 DFS로 해당 그룹들과 연결된 송전탑들이 더 이상 없을때 까지 탐색하면서 송전탑 그룹을 만든다.

4. 입력 받은 n에서 구해진 송전탑 그룹을 제하면 다른 송전탑 그룹이 나온다. 이둘의 차이를 구해서 반환한다.

5. 해당 경우 탐색이 끝나면 다시 기존의 추출했던 요소를 wires에 삽입하고 2번부터 반복한다.

6. 구해진 result들 가운데 가장 작은 수를 반환한다.

1. DFS

function solution(n, wires) { // 아직 미숙해서 성능상 효율적이지 않다.
    return Math.min(...wires.reduce((result, _) => {
        const current = wires.shift();
        result.push(checkWires(Array.from(wires)));
        wires.push(current);

        return result;
    }, []));

    function checkWires(wires) {
        const left = new Set(wires[0]);

        const dfs = (pre) => {
            let current = wires.findIndex((next) => next.includes(pre[0]) || next.includes(pre[1]));
            while (current !== -1) {
                const state = wires[current];
                left.add(state[0]);
                left.add(state[1]);
                wires.splice(current, 1);
                dfs(state);
                current = wires.findIndex((next) => next.includes(pre[0]) || next.includes(pre[1]));
            }
        }
        dfs(wires[0]);
        return Math.abs(n - left.size - left.size);
    }
}