문제
알고리즘 입문방 오픈 채팅방에서는 새로운 분들이 입장을 할 때마다 곰곰티콘을 사용해 인사를 한다. 이를 본 문자열 킬러 임스는 채팅방의 기록을 수집해 그 중 곰곰티콘이 사용된 횟수를 구해 보기로 했다.
ENTER는 새로운 사람이 채팅방에 입장했음을 나타낸다. 그 외는 채팅을 입력한 유저의 닉네임을 나타낸다. 닉네임은 숫자 또는 영문 대소문자로 구성되어 있다.
새로운 사람이 입장한 이후 처음 채팅을 입력하는 사람은 반드시 곰곰티콘으로 인사를 한다. 그 외의 기록은 곰곰티콘을 쓰지 않은 평범한 채팅 기록이다.
채팅 기록 중 곰곰티콘이 사용된 횟수를 구해보자!
입력
첫 번째 줄에는 채팅방의 기록 수를 나타내는 정수 N 이 주어진다. (1≤N≤100000)
두 번째 줄부터 N 개의 줄에 걸쳐 새로운 사람의 입장을 나타내는 ENTER, 혹은 채팅을 입력한 유저의 닉네임이 문자열로 주어진다. (문자열길이1≤문자열 길이≤20)
첫 번째 주어지는 문자열은 무조건 ENTER이다.
출력
채팅 기록 중 곰곰티콘이 사용된 횟수를 출력하시오.
예제
나의 풀이
import sys
input = sys.stdin.readline
N = int(input())
names = set()
cnt = 0
for i in range(N):
name = input().strip()
if name == 'ENTER':
cnt += len(names)
names = set()
else:
names.add(name)
cnt += len(names)
print(cnt)
- 새로운 사람이 등장한 후 처음으로 채팅을 입력한 사람들의 중복을 피하기 위해 `set()`을 사용했다.
- `ENTER`가 입력된 경우 그동안 채팅창에 곰곰티콘을 입력한 사람들의 수를 카운트해서 더하고, 새로 등장한 사람에게 인사한 사람들을 구하기 위해 `set()`을 초기화한다.