728x90
문제
알고리즘 입문방 오픈 채팅방에서는 새로운 분들이 입장을 할 때마다 곰곰티콘을 사용해 인사를 한다.
이를 본 문자열 킬러 임스는 채팅방의 기록을 수집해 그 중 곰곰티콘이 사용된 횟수를 구해 보기로 했다.
ENTER는 새로운 사람이 채팅방에 입장했음을 나타낸다.
그 외는 채팅을 입력한 유저의 닉네임을 나타낸다. 닉네임은 숫자 또는 영문 대소문자로 구성되어 있다.
새로운 사람이 입장한 이후 처음 채팅을 입력하는 사람은 반드시 곰곰티콘으로 인사를 한다.
그 외의 기록은 곰곰티콘을 쓰지 않은 평범한 채팅 기록이다.
채팅 기록 중 곰곰티콘이 사용된 횟수를 구해보자!
https://www.acmicpc.net/problem/25192
입력
첫 번째 줄에는 채팅방의 기록 수를 나타내는 정수 N 이 주어진다. (1 ≤ N ≤ 100,000)
두 번째 줄부터 N 개의 줄에 걸쳐 새로운 사람의 입장을 나타내는 ENTER, 혹은 채팅을 입력한 유저의 닉네임이 문자열로 주어진다.
(1 ≤ 문자열 길이 ≤ 20)
첫 번째 주어지는 문자열은 무조건 ENTER이다.
출력
채팅 기록 중 곰곰티콘이 사용된 횟수를 출력하시오.
예제
# input
9
ENTER
pjshwa
chansol
chogahui05
lms0806
pichulia
r4pidstart
swoon
tony9402
# output
8
# input
7
ENTER
pjshwa
chansol
chogahui05
ENTER
pjshwa
chansol
# output
5
문제 풀이
import sys
input = sys.stdin.readline
# 입력 받기
N = int(input())
# 현재 세션의 로그를 저장할 집합과 정답 변수 초기화
current_session_logs = set()
answer = 0
for _ in range(N):
log = input().rstrip()
if log == 'ENTER':
# 새로운 세션 시작, 현재 세션의 유니크 사용자 수를 정답에 더하고 초기화
answer += len(current_session_logs)
current_session_logs.clear()
else:
# 새로운 사용자의 로그를 집합에 추가
current_session_logs.add(log)
# 마지막 세션의 유니크 사용자 수를 정답에 더함
answer += len(current_session_logs)
print(answer)
파이썬을 독학하시는 분들에게 도움이 되길 바라며,
혹 더 좋은 방법이 있거나 오류가 있다면 편하게 말씀 부탁드립니다.
728x90