99클럽 코테 스터디 24일차 TIL : 완전탐색(프로그래머스 - 전력망을 둘로 나누기)

 

 

🔎 문제 

 

 

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

프로그래머스

 

---

💡 풀이

- DFS

- 배열로 된 트리 좌표들을 LIST배열을 만들어서 저장 ex) 노드 1의 연결된 좌표는 wireList[1] 에 저장

- divideNetwork(n, wireList, wires[i]) : 주어진 wires 이차배열을 하나씩 가지고 오면서 해당 연결(wires[i])을 해제

    1. wireList에서 끊은 연결 좌표를 제거 ex) wireList[edge[0]].remove((Object)edge[1]);

    2. 끊은 노드 각각에서부터 dfs 출발  ex ) dfs(n, wireList, new boolean[n+1], edge, edge[0], true);

    3. cnt1, cnt2에 각각 두부분으로 나눠진 그룹들이 몇개의 노드가 연결되어있는지를 저장
    4. 이후 cnt1, cnt2를 뺀 절대값중에 최소값을 저장

 

import java.util.ArrayList;
import java.util.List;

class Solution {
    public int solution(int n, int[][] wires) {
  
        List<Integer>[] wireList = new ArrayList[n+1]; 
		
		for(int i=1;i<=n;i++) {
			wireList[i] = new ArrayList<Integer>();
		}
		
		for(int i=0;i<wires.length;i++) {
				wireList[wires[i][0]].add(wires[i][1]);
				wireList[wires[i][1]].add(wires[i][0]);
		}
		
		for(int i=0;i<wires.length;i++) {
			divideNetwork(n, wireList, wires[i] );
		}
        
        return min;
    }
    
    private int cnt1= 0;
	private int cnt2= 0;
    private int min= Integer.MAX_VALUE;
	
	private void divideNetwork(int n, List<Integer>[] wireList, int[] edge) {
		wireList[edge[0]].remove((Object)edge[1]);
		wireList[edge[1]].remove((Object)edge[0]);
		
		dfs(n, wireList, new boolean[n+1], edge, edge[0], true);
		dfs(n, wireList, new boolean[n+1], edge, edge[1], false);
		
		int absValue = Math.abs(cnt1-cnt2);
        min = Math.min(absValue,min);
        
        cnt1 = 0;
        cnt2 = 0;
        wireList[edge[0]].add(edge[1]);
		wireList[edge[1]].add(edge[0]);
		
	}
	
	private void dfs(int n, List<Integer>[] wireList, boolean[] visited, int[] edge, int k, boolean isCnt1) {
		visited[k] = true;
        
		if (isCnt1) {
            cnt1++;
        } else {
            cnt2++;
        }

		for(int node : wireList[k]) {
			if(visited[node]==false) {
				dfs(n,wireList,visited,edge,node,isCnt1);
			}
		}
	}
}

 

 

 

 

📌 오늘의 회고

늦게 시작했더니 제출을 시간안에 못해서 끝나버렸다 ,,, 

그래도 하던거 해결하고 싶어서 넘 답답해서 끝까지 했지만 🥺🥺🥺🥺🥺🥺

아쉽다아