[ DFS / BFS ] - 네트워크(DFS, BFS)

문제

네트워크란 컴퓨터 상호 간에 정보를 교환할 수 있도록 연결된 형태를 의미합니다. 예를 들어, 컴퓨터 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입니다.

입출력 예

n	computers	return
3	[[1, 1, 0], [1, 1, 0], [0, 0, 1]]	2
3	[[1, 1, 0], [1, 1, 1], [0, 1, 1]]	1

입출력 예 설명

예제 #1
아래와 같이 2개의 네트워크가 있습니다.

예제 #2
아래와 같이 1개의 네트워크가 있습니다.

 

풀이

1. 완전 탐색 문제이므로 DFS, BFS 어느걸로 풀어도 상관이 없다.

2. 간선이 이루어진 노드 갯수와 떨어져 있는 노드 수를 합치면 된다.
3. 전체를 탐색하면서 방문한 적이 없고(false && computers[][] == 1)인 경우 재귀 혹은 큐에 인접한 값을 반복 실행하며 방문한 적이 없는 경우에 +1 해준다.

 

코드

BFS

import java.util.LinkedList;
import java.util.Queue;
class Solution {
    // 자료구조 
    int count = 0;
    boolean[] visit;
    Queue<Integer> q;
    public int solution(int n, int[][] computers) {
        int answer = 0;
        visit = new boolean[computers.length];
        // BFS 호출
        // 컴퓨터 갯수 만큼 반복 
        for(int i=0; i<n; i++){
            if(visit[i] == false){
                bfs(i,computers,n);
                count++;
            }
        }
        answer = count;            
        return answer;
    }
    public void bfs(int start, int[][] computers, int n){ 
        q = new LinkedList<Integer>();
        q.offer(start);
        // 방문체크
        visit[start] = true;
        while(!q.isEmpty()){
            int x = q.poll();
            for(int j=0; j<n; j++){
                // 방문 안하고 큐에 담긴 값의 배열값이 1인 경우 방문 처리
                if(visit[j] == false && computers[x][j] == 1){
                    visit[j] = true;
                    q.offer(j);
                }
            }
        }
    }
}

 

DFS

class Solution {
    // 자료구조 
    int count = 0;
    boolean[] visit;
    
    public int solution(int n, int[][] computers) {
        int answer = 0;
        visit = new boolean[computers.length];

        for(int i=0; i<n; i++){
            if(visit[i] == false){
                dfs(i,computers,n);
                count++;
            }
        }
        answer = count;            
        return answer;
    }
    public void dfs(int start, int[][] computers, int n){ 

        // 방문체크
        visit[start] = true;
        for(int j=0; j<n; j++){
            if(visit[j] == false && computers[start][j] == 1){
                dfs(j,computers, n);
            }
        }
    }
}