[Algorithm] 백준 25192번 인사성 밝은 곰곰이 (Python)

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

25192번: 인사성 밝은 곰곰이

문제

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

ENTER는 새로운 사람이 채팅방에 입장했음을 나타낸다. 그 외는 채팅을 입력한 유저의 닉네임을 나타낸다. 닉네임은 숫자 또는 영문 대소문자로 구성되어 있다.

새로운 사람이 입장한 이후 처음 채팅을 입력하는 사람은 반드시 곰곰티콘으로 인사를 한다. 그 외의 기록은 곰곰티콘을 쓰지 않은 평범한 채팅 기록이다.

채팅 기록 중 곰곰티콘이 사용된 횟수를 구해보자!

입력

첫 번째 줄에는 채팅방의 기록 수를 나타내는 정수 N이 주어진다. (1 ≤ N ≤ 100000)

두 번째 줄부터 N개의 줄에 걸쳐 새로운 사람의 입장을 나타내는 ENTER, 혹은 채팅을 입력한 유저의 닉네임이 문자열로 주어진다. (1 ≤ 문자열 길이 ≤ 20)

첫 번째 주어지는 문자열은 무조건 ENTER이다.

출력

채팅 기록 중 곰곰티콘이 사용된 횟수를 출력하시오.

예제 입력 1

9
ENTER
pjshwa
chansol
chogahui05
lms0806
pichulia
r4pidstart
swoon
tony9402

예제 출력 1

8

예제 입력 2

7
ENTER
pjshwa
chansol
chogahui05
ENTER
pjshwa
chansol

예제 출력 2

5

첫 번째 새로운 사람이 들어온 뒤  pjshwa, chansol, chogahui05은 모두 곰곰티콘으로 인사했다.

두 번째 새로운 사람이 들어온 뒤  pjshwa와 chansol은 다시 곰곰티콘으로 인사했다.

예제 입력 3

3
ENTER
lms0806
lms0806

예제 출력 3

1

lms0806은 새로운 사람이 들어왔으므로 처음은 곰곰티콘으로 인사하고, 그 뒤로는 일반 채팅을 했다.

풀이

설명

• 리스트로 접근하면 시간초과가 발생한다.
• 입력받은 단어가 ENTER이면 user 집합 안의 요소들을 지우고 아니면 user에 이름을 추가하여 cnt를 1만큼 증가한다.

코드

import sys

input = sys.stdin.readline

N = int(input())
user = set()
cnt = 0

for i in range(N):
    txt = str(input().rstrip())

    if txt == "ENTER":
        user.clear()
    else:
        if txt not in user:
            user.add(txt)
            cnt += 1

print(cnt)