Algorithm 🧑🏻‍💻/CodeUp

4503 : 바이러스

dkswnkk 2022. 1. 14. 22:59

문제

 

바이러스

1번 컴퓨터가 웜 바이러스에 걸렸을 때, 1번 컴퓨터를 통해 웜 바이러스에 걸리게 되는 컴퓨터의 수를 첫째 줄에 출력한다.

codeup.kr

 

코드

#include <iostream>
#include <vector>

using namespace std;

int N,M,ans;
int visited[101];
vector<int>v[101];

void dfs(int start){
    visited[start]=1;
    ans++;
    for(int i=0; i<v[start].size(); i++){
        if(visited[v[start][i]]) continue;
        dfs(v[start][i]);
    }
}
int main(){
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    
    
    cin>>N>>M;
    for(int i=0; i<M; i++){
        int a,b; cin>>a>>b;
        v[a].push_back(b);
        v[b].push_back(a);
    }
    dfs(1);
    cout<<ans-1;
}

 

채점