[C++] 백준 26069. 붙임성 좋은 총총이

문제

풀이

저번 곰곰이 문제랑 비슷한데 살짝 더 쉬운 버전이다.
새로 선언한 set에 댄싱머신 총총이를 미리 넣어주고
새로 받은 두 이름 중 하나라도 댄싱머신 set에 있다면
다른 이름을 push한다.
이미 댄싱머신인 사람은 들어갈 필요가 없으니 자료구조 set을 썼다.
마지막에 set의 size를 출력해주면 댄싱머신의 숫자를 알 수 있다.

코드

#include <iostream>
#include <set>
#include <string>
using namespace std;

int main() {
	int n;
	cin >> n;

	set<string> s;
	s.insert("ChongChong");

	string s1, s2;
	for (int i = 0; i < n; i++) {
		cin >> s1 >> s2;
		if (s.find(s1) != s.end()) {
			s.insert(s2);
		}
		else if (s.find(s2) != s.end()) {
			s.insert(s1);
		}
	}
	cout << s.size();

	return 0;
}

채점