[프로그래머스] 네트워크 / BFS - Lv.3

https://programmers.co.kr/learn/courses/30/lessons/43162

 

문제 설명

네트워크란 컴퓨터 상호 간에 정보를 교환할 수 있도록 연결된 형태를 의미합니다. 예를 들어, 컴퓨터 A와 컴퓨터 B가 직접적으로 연결되어있고, 컴퓨터 B와 컴퓨터 C가 직접적으로 연결되어 있을 때 컴퓨터 A와 컴퓨터 C도 간접적으로 연결되어 정보를 교환할 수 있습니다. 따라서 컴퓨터 A, B, C는 모두 같은 네트워크 상에 있다고 할 수 있습니다.

컴퓨터의 개수 n, 연결에 대한 정보가 담긴 2차원 배열 computers가 매개변수로 주어질 때, 네트워크의 개수를 return 하도록 solution 함수를 작성하시오.

 

제한사항

• 컴퓨터의 개수 n은 1 이상 200 이하인 자연수입니다.
• 각 컴퓨터는 0부터 n-1인 정수로 표현합니다.
• i번 컴퓨터와 j번 컴퓨터가 연결되어 있으면 computers[i][j]를 1로 표현합니다.
• computer[i][i]는 항상 1입니다.

 

문제 해결 아이디어

 

처음에는 배열 보고서 "가운데(computers[i][i]) 기준으로 양쪽이 같으니까 한쪽만 돌면 되겠네 ㅎㅎ" 하고 이중for문으로 i는 0부터 n-1까지, j는 i+1부터 n-1까지 해서 신나게 bfs 돌렸다. 안됨!

내가 생각했던 테케는 다 됐는데 제출하면 안 됐다. 그래서 테케 막 생각해 보다가 그래프가 좀 덜 예쁘게 연결된? 케이스를 생각해 냈다. 만약 2번, 5번이 연결되어 있고 3번, 4번이 연결되어있는 상황에서 5번이랑 4번을 연결해 버리면? 2와 연결된 5에서 5와 연결된 4에 도달하지 못하는 상황이 발생한 것이다.

그래서 바로 생각을 수정해서 그냥 모든 경우에 대해 bfs 했다.

방문하지 않은 컴퓨터마다 bfs를 돌아서 같은 그래프 안에 있으면 같은 네트워크 상에 있는 것이다. 그랬더니 해결!

 

평소에 히든 테케를 찾는 걸 어려워하는 경향이 있는데 이건 그래도 비교적 빠른 시간 내에 찾아서 다행이다 ㅠㅋㅋ

 

import java.util.*;

class Solution {
    static boolean[] visited;
    
    public int solution(int n, int[][] computers) {
        visited = new boolean[n];
        int answer = 0;
        
        for (int i = 0; i < n; i++) {
                if (!visited[i]) {
                    visited[i] = true;
                    bfs(i, n, computers);
                    answer++;
                }
        }
        return answer;
    }
    
    public void bfs(int i, int n, int[][] computers) {
        Queue<Integer> queue = new LinkedList<>();
        queue.offer(i);
        
        while (queue.size() > 0) {
            int now = queue.poll();
            
            for (int c = 0; c < n; c++)  {
                if (c != now && computers[now][c] == 1 && !visited[c]) {
                    visited[c] = true;
                    queue.offer(c);
                }
            }
        }
    }
    
}