[백준/BOJ] 2606번 - 바이러스 (C++)

문제 링크

코드

#include <bits/stdc++.h>
using namespace std;
vector<int> p[101];
bool vis[101];
int n, t, a, b, cnt = 0;

void dfs(int x) {
    vis[x] = 1;
    for (int i = 0; i < p[x].size(); i++) {
        int y = p[x][i];
        if (!vis[y]) {
            dfs(y);
            cnt++;
        }
    }
}

int main(void) {
    ios::sync_with_stdio(0);
    cin.tie(0);
    cin >> n >> t;
    while (t--) {
        cin >> a >> b;
        p[a].push_back(b);
        p[b].push_back(a);
    }
    dfs(1);
    cout << cnt << '\n';
}

---

설명

DFS 방식으로 해결하였다.

(주의) 기록용으로 작성한 글입니다. 좋은 코드가 아닐 수 있습니다.

댓글 환영합니다!