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;
}