[BOJ] 2606. 바이러스

문제

www.acmicpc.net/problem/2606

2606번: 바이러스

---

해설

---

코드

#include<iostream>
#include<queue>
#include<algorithm>

using namespace std;

int map[101][101];
int visit[101];
int cnt;
int n, com; // 컴퓨터의 수, 네트워크 연결 쌍


void dfs(int node) {

	visit[node] = 1;

	for (int i = 1; i <= n; i++) {
		if (map[node][i] == 1 && !visit[i]) {
			cnt++;
			dfs(i);
		}
	}
}

void bfs(int node) {

	visit[node] = 1;
	queue<int> q;
	q.push(node);

	while (!q.empty()) {
		node = q.front();
		q.pop();

		for (int i = 1; i <= n; i++) {
			if (map[node][i] == 1 && !visit[i]) {
				q.push(i);
				visit[i] = 1;
				cnt++;
			}
		}
	}
}

int main() {

	ios::sync_with_stdio(false);
	cin.tie(0), cout.tie(0);
	int x, y;
	cin >> n;
	cin >> com;
	for (int i = 0; i < com; i++) {
		cin >> x >> y;
		map[x][y] = map[y][x] = 1;
	}

	dfs(1);
	cout << cnt << '\n';
	cnt = 0;
	memset(visit, 0, sizeof(visit));
	bfs(1);
	cout << cnt << '\n';

}