Lv2. 전력망을 둘로 나누기

* Programmers _위클리 챌린지 문제입니다.

* 언어는 javascript 를 선택했습니다.

 

1. 문제

1) 전력망을 둘로 나누기

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

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

 

2) 제한사항

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

 

 

2. 알고리즘! 생각해보자

1. 전선 하나씩 없애가면서 분할된 네트워크 bfs로 찾기
=> 이 때, 자료구조는 {idx : [[network_01], [network_02]], ...}
2. 분할된 네트워크 중 송전탑 개수 최소인 경우 return

(주의할 점)

1. 전선 하나씩 없애가면서 분할된 네트워크 bfs로 찾기

• 시작지점 : 삭제한 전선이 시작지점이 된다
• return 조건 : 분할된 네트워크 내에 추가할 item 이 존재하면, bfs stop!

 

 

3. 해결 코드

function solution(n, wires) {
    var answer = [];
    var testNetwork = {}; // {idx : [[network_01], [network_02]]}
    
    function bfsNetwork(networkKey, isFirstNetwork, wires) {
        var networkIdx = isFirstNetwork ? 0 : 1;
        var networkLen = testNetwork[networkKey][networkIdx].length - 1;
        var lastItem = testNetwork[networkKey][networkIdx][networkLen];
        var nextItems = wires.filter(v => v.includes(lastItem));
        nextItems.length && nextItems.forEach((v, idx) => {
            var addedItem = (lastItem === v[0]) ? v[1] : v[0];
            if(testNetwork[networkKey][networkIdx].includes(addedItem)) {
                return;
            }
            testNetwork[networkKey][networkIdx].push(addedItem);
            var removedWires = wires.filter(wire => !(wire[0] === v[0] && wire[1] === v[1]));
            bfsNetwork(networkKey, isFirstNetwork, removedWires);
        });
    }
    
    // 1. 전선 하나씩 없애가면서 분할된 네트워크 bfs로 찾기
    wires.forEach((wire, idx)=> {
        testNetwork[idx] = [[], []];
        var removedWire = wires.filter(v => !(wire[0] === v[0] && wire[1] === v[1]));
        testNetwork[idx][0].push(wire[0]);
        testNetwork[idx][1].push(wire[1]);  
        bfsNetwork(idx, true, removedWire); // 첫번째 네트워크 망
        bfsNetwork(idx, false, removedWire); // 두번째 네트워크 망
    });
    
    // 2. 분할된 네트워크 중 송전탑 개수 최소인 경우 return
    for(var item in testNetwork) {
        answer.push(Math.abs(testNetwork[item][0].length-testNetwork[item][1].length));
    }
    return Math.min(...answer);
}

 

 

4. 문제해결 능력 UP! 되짚어보기

처음에는 bfsNetwork() 내에 첫번째, 두번째 네트워크 각각을 bfs 해주고 있어서 코드량이 2배였습니다.

그런데 문제를 풀고나니 굳이 같은 코드를 반복할 필요가 없겠더라구요! 

그래서 리팩토링 해주었습니다. 초기 코드는 아래와 같습니다~

function bfsNetwork(key, wires) {
        var lastItem1 = testNetwork[key][0][testNetwork[key][0].length - 1];
        var lastItem2 = testNetwork[key][1][testNetwork[key][1].length - 1];
        var nextItem1 = wires.filter(v => v.includes(lastItem1));
        nextItem1.length && nextItem1.forEach((v, idx) => {
            var addedItem = (lastItem1 === v[0]) ? v[1] : v[0];
            if(testNetwork[key][0].includes(addedItem)) {
                return;
            }
            testNetwork[key][0].push(addedItem);
            var what = wires.filter(wire => !(wire[0] === v[0] && wire[1] === v[1]));
            bfsNetwork(key, what);
        });
        
        var nextItem2 = wires.filter(v => v.includes(lastItem2));
        nextItem2.length && nextItem2.forEach(v => {
            var addedItem = (lastItem2 === v[0]) ? v[1] : v[0];
            if(testNetwork[key][1].includes(addedItem)) {
                return;
            }
            testNetwork[key][1].push(addedItem);
            var what = wires.filter(wire => !(wire[0] === v[0] && wire[1] === v[1]));
            bfsNetwork(key, what);
        });
    }

 

 

5. References