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

문제 풀러가기

문제

난이도: 레벨 2

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

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

입출력 예
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

---

문제 분석

완전탐색 문제
트리라는 자료구조를 오랜만에 들어서 당황했지만 그래도 잘 풀었다!
(트리는 계층형을 나타내는 비선형 자료구조)

하나의 트리 형태로 이루어져 있기 때문에 주어진 wires 중 하나만 끊어도 두개로 나뉘게 된다

이점을 활용해서 모든 wires 개수대로 끊어주면 됨!

---

풀이 방법

wires.length만큼 반복해서 bfs 돌려주기

돌릴 때마다 트리 구조를 나타내는 2차원 리스트를 초기화 해주는데,
여기서 현재 제외할(끊을 선) idx의 wires는 제외하고 초기화 해준다

그리고 나서 임의로 정점 0번을 넣어 bfs를 돌린 후,
방문한 곳과 방문하지 않은 곳으로 나뉘게 되는데 카운트 해줘서 계산해주면 된다

---

내 코드

import java.util.*;

class Solution {
    private static int answer;
    private static boolean[] visit;
    private static List<List<Integer>> list;
    
    public int solution(int n, int[][] wires) {
        answer = 987654321;
        
        for(int i = 0; i<wires.length; i++) {
            bfs(i, n, wires); //끊을 선
        }
        
        return answer;
    }
    
    private static void bfs(int idx, int n, int[][] wires) {
        
        list = new ArrayList<>();
        for(int i = 0; i<n; i++) {
            list.add(new ArrayList<>());
        }
        
        for(int i = 0; i<wires.length; i++) {
            if(i == idx) continue;
            int a = wires[i][0];
            int b = wires[i][1];
            
            list.get(a-1).add(b-1);
            list.get(b-1).add(a-1);
        }
        
        Queue<Integer> q = new ArrayDeque<>();
        q.offer(0);
        
        visit = new boolean[n];
        visit[0] = true;
        
        while(!q.isEmpty()) {
            int cur = q.poll();
            
            for(int o: list.get(cur)) {
                if(!visit[o]) {
                    visit[o] = true;
                    q.offer(o);
                }
            }
        }
        //System.out.println(Arrays.toString(visit));
        
        int one = 0, two = 0;
        for(int i = 0; i<n; i++) {
            if(visit[i]) one++;
            else two++;
        }
        //System.out.println(one + " " + two);
        answer = Math.min(answer, Math.abs(one-two));
        //System.out.println(answer);
    }
}