백준 26069 붙임성 좋은 총총이 / C++

문제 : 26069

풀이 point

ChongChong이와 만난 이후에 사람들만 따라서 춤을 춘다.
총총이와 만난 이후 사람들을 확인하고 숫자를 세준다.

풀이 방법

map or set을 활용하여 풀 수 있다.
참고로 ChongChong이를 미리 자료구조에 넣어주는 것이 계산하기 좋다.

코드

//백준 26069, 붙임성 좋은 총총이
#include <iostream>
#include <vector>
#include <set>

std::set<std::string> set;
std::set<std::string> visit;

int main(){

    int N;
    std::cin >> N;
    std::string a; std::string b;
    int ans{0};
    set.insert("ChongChong");
    for(int i{0}; i<N; ++i){
        std::cin >> a >> b;
        if((visit.find(b) == visit.end()) && (set.find(a) != set.end())){
            set.insert(b);
            visit.insert(b);
        }
        if((visit.find(a) == visit.end()) && (set.find(b) != set.end())){
            set.insert(a);
            visit.insert(a);
        }
        if(a == "ChongChong" && set.find(b) == set.end()){
            set.insert(b);
            visit.insert(b);
        }
        if(b == "ChongChong" && set.find(a) == set.end()){
            set.insert(a);
            visit.insert(a);
        }
    }
    
    std::cout << set.size();

    return 0;
}