[BOJ 25192 // C++] 인사성 밝은 곰곰이

※ 글쓴이는 취미로 코딩을 익혀보는 사람이라 정확하지 않은 내용을 담고 있을 수 있다 ※

 

이번에 볼 문제는 백준 25192번 문제인 인사성 밝은 곰곰이이다.
문제는 아래 링크를 확인하자.

https://www.acmicpc.net/problem/25192 

25192번: 인사성 밝은 곰곰이

"ENTER"이 입력된 이후, 다음 "ENTER"이 입력될 때까지 등장하는 문자열의 가짓수만큼씩을 답에 추가해주자.

 

set을 이용하면 등장한 문자열들을 편하게 관리할 수 있다.

 

특히, 주어지는 입력에 이어 추가로 마지막에 "ENTER"이 입력되어도 답이 변하지 않으므로, 마지막에 "ENTER"이 주어졌다고 생각하고 구현한다면 답을 편하게 구할 수 있다.

 

아래는 제출한 소스코드이다.

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

set<string> st;

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);

	int ans = 0;
	int N; cin >> N;
	while (N--) {
		string s; cin >> s;
		if (s == "ENTER") {
			ans += st.size();
			st.clear();
		}
		else st.insert(s);
	}

	ans += st.size();
	cout << ans;
}