[c++][BOJ][2606] 바이러스

https://www.acmicpc.net/problem/2606

2606번: 바이러스

1번 컴퓨터가 바이러스에 걸렸을 때 , 연결된 컴퓨터 몇 대가 감염되는지 구하는 문제입니다.

 

먼저 그래프 세팅을 하고 , 1번과 연결되어있는 컴퓨터를 찾습니다.

 

감염된 컴퓨터를 체크하고 , 연결된 컴퓨터에서 다시 연결된 컴퓨터를 찾습니다.

 

마지막에 감염된 컴퓨터의 수를 구하고 1을 제거합니다.(1번 컴퓨터)

#include <iostream>
using namespace std;

int arr[101][101] = {0};
int check[101] = {0};

void find(int start, int index, int max){
    arr[start][index] = 0;
    arr[index][start] = 0;
    
    check[index] = 1;
    
    for(int i=1; i<=max; i++){
        if(arr[index][i] == 1){
            find(index, i, max);
        }
    }
}

int main(){
    int n;
    cin >> n;

    int m;
    cin >> m;
    
    for(int i=0; i<m; i++){
        int x, y;
        cin >> x >> y;
        
        arr[x][y] = 1;
        arr[y][x] = 1;
    }
    
    check[1] = 1;
    for(int i=1; i<=n; i++){
        if(arr[1][i] == 1){
            find(1, i, n);
        }
    }
    int count = 0;
    
    for(int i=1; i<=n; i++){
        if(check[i] == 1)
            count++;
    }
    
    cout << count-1;
}