그래프 순회 응용 (네트워크)

문제 설명

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

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

 

 

∙문제 해결 방법

- 그래프 순회(DFS, BFS)를 이용하여 해결 가능

- 그래프의 정점을 모두 방문하기 위해 필요한 (최소) 그래프 순회 횟수를 카운트

- 즉, 0부터 n - 1까지 정점 번호를 증가시키면서 해당 정점을 아직 방문하지 않았다면 DFS 또는 BFS 알고리즘을 수행하고 그 수행 횟수를 카운트하여 반환

 

#include <iostream>
#include <string>
#include <vector>
#include <queue>

using namespace std;

int solution(int n, vector<vector<int>> computers)
{
    vector<bool> visited(n, false);
    int num_networks = 0;
    for (int i = 0; i < n; i++){
        // BFS
        
        if (visited[i])
            continue;
        
        queue<int> q;
        q.push(i);
        
        while (!q.empty()){
            int v = q.front();
            q.pop();
            
            if (visited[v])
                continue;
            
            visited[v] = true;
            
            for (int j = 0; j < n; j++){
                if (v !=j && computers[v][j] == 1 && !visited[j])
                    q.push(j);
            }
        }
        num_networks++;
    }
    return num_networks;
}

int main()
{
    cout <<solution(3,{{1, 1, 0}, {1, 1, 0}, {0, 0, 1}}) << endl; // 2
    
    cout <<solution(3,{{1, 1, 0}, {1, 1, 1}, {0, 1, 1}}) << endl; // 1
    
}