(BOJ-26069) 아군 총

이는 지도를 통해 모두가 무지개 춤을 추는지 여부를 관리하면 쉽게 해결할 수 있습니다.


#include <iostream>
#include <string>
#include <map>

using namespace std;

int N;
string A, B;
map<string, bool> mp;

void solve() {
    mp("ChongChong") = true;
    cin >> N;
    for (int i = 0; i < N; i++) {
        cin >> A >> B;
        if (mp(A) || mp(B)) {
            mp(A) = true;
            mp(B) = true;
        }
    }
    
    int sum = 0;
    for (auto p : mp) {
        sum += p.second;
    }
    
    cout << sum << "\n";
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);
    
    solve();
}