[백준] 25192번 인사성 밝은 곰곰이 | 파이썬 Python

문제

알고리즘 입문방 오픈 채팅방에서는 새로운 분들이 입장을 할 때마다 곰곰티콘을 사용해 인사를 한다. 이를 본 문자열 킬러 임스는 채팅방의 기록을 수집해 그 중 곰곰티콘이 사용된 횟수를 구해 보기로 했다.

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()`을 초기화한다.