[백준] 25192번 : 인사성 밝은 곰곰이(실버Ⅳ) - Python

문제

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

25192번: 인사성 밝은 곰곰이

문제

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

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

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

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

입력

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

두 번째 줄부터 
$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

예제 입력 3

 3
ENTER
lms0806
lms0806

예제 출력 3

 1

 

풀이

접근

1. 새로운 사람이 들어온 이후에 처음 채팅을 친 사람의 이름을 저장할 set를 만든다.
2. ENTER -> 새로운 사람이 들어오면 해당 set를 초기화한다.
3. 그 다음 나온 사람의 이름이 set에 없으면 set에 추가하고,
4. gomgom 이모티콘 사용 횟수 += 1

풀이 코드

import sys
input = sys.stdin.readline

N = int(input())

new_chat = set()
cnt_gomgom = 0

for _ in range(N):
    chat = input().rstrip()
    
    if chat == 'ENTER': # 새로운 사람이 들어오면
        new_chat.clear() # 채팅 정보 초기화
            
    elif chat != 'ENTER':
        if chat not in new_chat: # 해당 사용자가 처음 채팅을 보냈다면
            new_chat.add(chat)
            cnt_gomgom+= 1
        
print(cnt_gomgom)