[프로그래머스/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입니다.

풀이

해당 문제는 깊이/너비 우선 탐색 (DFS/BFS) 문제입니다.

저는 이 문제는 DFS(깊이 우선 탐색)로 풀었는데요.

 

우선 network라는 (dfs 함수에서는 visited) 해당 network를 확인했는지 표시하는 boolean형 배열을 선언했습니다. netwok의 해당 인덱스가 비어있을 경우, dfs 함수에서 해당 네트워크와 연결된 네트워크를 탐색하도록 작성했습니다.

 

dfs 함수는 해당 네트워크에 연결된 네트워크 중에서 (computer[node][idx] == 1)

본인 인덱스가 아니고 (node != idx)

아직 확인 안 된 네트워크일 경우(visited[idx] == false)

dfs 함수가 실행되도록 작성했습니다.

코드

import java.util.HashSet;

class Solution {
    public int solution(int n, int[][] computers) {
        int answer = 0;
        boolean[] network = new boolean[n];
        
        for (int node=0; node<n; node++) {
            if (!network[node]) {
                dfs(computers, node, network);
                answer++;
            }
        }
        
        return answer;
    }
    
    boolean[] dfs(int[][] computers, int node, boolean[] visited) {
        visited[node] = true;
        
        for (int idx=0; idx<computers.length; idx++) {
            if (node != idx && computers[node][idx] == 1 && visited[idx] == false) {
                visited = dfs(computers, idx, visited);
            }
        }
        return visited;
    }
}

채점 결과