알고리즘 풀이/프로그래머스

[프로그래머스][JAVA] 9주차_전력망을 둘로 나누기 (그래프, DFS)

배게 2021. 10. 29. 17:34
728x90

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
import java.util.*;
 
class Solution {
    List<Node> nodeList;
    int nodeNum;
    int answer;
    boolean[] visited;
    
    class Node{
        int num;
        List<Integer> edge;
        
        Node(int num){
            this.num = num;
            edge = new ArrayList<>();
        }
        
    }
    
    public int solution(int n, int[][] wires) {
        nodeList = new ArrayList<>();
        answer = n;
        visited = new boolean[n+1];
        
        for(int i=0; i<=n; i++)
            nodeList.add(new Node(i));
        
        
        for(int[] wire : wires){
            int a = wire[0];
            int b = wire[1];
            nodeList.get(a).edge.add(b);
            nodeList.get(b).edge.add(a);
        }
        
        
        
        // wire끊기
        for(int[] wire : wires){
            Integer a = new Integer(wire[0]);
            Integer b = new Integer(wire[1]);
            nodeList.get(a).edge.remove(b);
            nodeList.get(b).edge.remove(a);
            
            // a에서 시작
            nodeNum = 0;
            DFS(a);
            // System.out.println("nodeNum^^ : "+ nodeNum);
            
            answer = Math.min(answer, Math.abs(2*nodeNum-n));
            
            nodeList.get(a).edge.add(b);
            nodeList.get(b).edge.add(a);
        }
            
        
        
        return answer;
    }
    
    private void DFS(int currNum){
        Node curr = nodeList.get(currNum);
        nodeNum++;
        visited[currNum] = true;
        
        for(int nodeNum : curr.edge){
            if(!visited[nodeNum]){
                DFS(nodeNum);
            }
            
        }
        
        
        visited[currNum] = false;
        
    }
}
cs