[level 2] Title: 전력망을 둘로 나누기, Time: 9.16 ms, Memory: 75 MB -BaekjoonHub

Author: wonseokyoon

프로그래머스/2/86971. 전력망을 둘로 나누기/README.md

@@ -4,7 +4,7 @@
 
 ### 성능 요약
 
-메모리: 74 MB, 시간: 8.99 ms
+메모리: 75 MB, 시간: 9.16 ms
 
 ### 구분
 
@@ -16,7 +16,7 @@
 
 ### 제출 일자
 
-2025년 09월 05일 11:44:16
+2025년 09월 06일 19:55:10
 
 ### 문제 설명
 

프로그래머스/2/86971. 전력망을 둘로 나누기/전력망을 둘로 나누기.java

@@ -1,44 +1,44 @@
 import java.util.*;
+
 class Solution {
     boolean[] visited;
-    int answer;
     public int solution(int n, int[][] wires) {
-        answer = n;
         visited = new boolean[n+1];
-        
-        for(int[] wire : wires){
+        int answer = n;
+        int cnt = 1;
+        for(int[] wire : wires) {
             int towerA = wire[0];
             int towerB = wire[1];
-            visited[towerB] = true;
+
+            // towerA와 연결된 타워의 갯수
             visited[towerA] = true;
-            int cnt = dfs(towerA,wires,n);  // towerA와 연결된 타워만 찾으면 됨
+            visited[towerB] = true;
+            cnt = dfs(towerA,wires);
             visited[towerA] = false;
             visited[towerB] = false;
 
-            answer = Math.min(Math.abs(n - (2 * cnt)), answer);
+            answer = Math.min(Math.abs((n - cnt) - cnt),answer);
         }
-        
         return answer;
     }
 
-    int dfs(int towerA, int[][] wires, int n){
+    public int dfs(int towerA, int[][] wires){
         int cnt = 1;
-        for(int[] wire : wires){
+        
+        for(int i = 0; i < wires.length; i++) {
+            int[] wire = wires[i];
 
-            // 연결된 송전탑 찾음(방문한적 x)
-            if(wire[0] == towerA && !visited[wire[1]]) {
-                visited[wire[1]] = true;    // 방문 처리
-                cnt += dfs(wire[1],wires,n); // dfs
-                visited[wire[1]] = false;   // 백트래킹
-
-            } else if(wire[1] == towerA && !visited[wire[0]]) {
-                visited[wire[0]] = true;    // 방문 처리
-                cnt += dfs(wire[0],wires, n); // dfs
-                visited[wire[0]] = false;   // 백트래킹
+            if(towerA == wire[0] && !visited[wire[1]]){
+                visited[wire[1]] = true;
+                cnt += dfs(wire[1], wires);
+                visited[wire[1]] = false;
+            } else if(towerA == wire[1] && !visited[wire[0]]){
+                visited[wire[0]] = true;
+                cnt += dfs(wire[0], wires);
+                visited[wire[0]] = false;
             }
         }
 
         return cnt;
     }
-    
 }