[Baekjoon] 2606번: 바이러스

🥅 문제

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

 

🔍 설계

1번과 연결된 컴퓨터들이 감염된다는 것에서 BFS를 활용해야 한다는 것을 알 수 있다. 방문 처리를 하여 중복해서 큐에 담지 않도록 설계하였다.

 

👊 풀이

#include <iostream>
#include <vector>
#include <queue>
#include <algorithm>
#include <cmath>
using namespace std;	
int n, m, a, b, result=0;
vector<vector<int>> adj_list(101);
vector<bool> visited(101);
queue<int> q;

void bfs(){
	visited[1] = true;
	q.push(1);
	while (!q.empty()){
		int x = q.front();
		q.pop();
		for (int i=0; i<adj_list[x].size(); i++){
			int y = adj_list[x][i];
			if (!visited[y]){
				q.push(y);
				visited[y] = true;
				result += 1;
			}
		}
	}
}

int main() {
	cin.tie(NULL);
	ios_base::sync_with_stdio(false);
	cin >> n >> m;
	for (int i=0; i<m; i++){
		cin >> a >> b;
		adj_list[a].push_back(b);
		adj_list[b].push_back(a);
	}
	bfs();
	cout << result << "\n";

	return 0;
}