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

문제

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

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

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

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

입력

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

두 번째 줄부터
$N$ 개의 줄에 걸쳐 새로운 사람의 입장을 나타내는 ENTER, 혹은 채팅을 입력한 유저의 닉네임이 문자열로 주어진다. (문자열길이
$1 \le \texttt{문자열 길이} \le 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 - 새로운 사람 입장
# 그 외 채팅 입력한 유저 닉네임 
# 닉네임 - 숫자 or 영문 대소문자
# 새로운 사람 입장 후 처음 채팅 입력하는 사람 반드시 곰곰티콘으로 인사
# 그외의 기록은 곰곰티콘 사용x 평범한 채팅 기록
# 곰곰티콘 사용된 횟수
# 첫째줄 채팅방 기록수 N
# 둘째줄부터 N 개의 줄에 걸쳐 새로운 사람의 입장을 나타내는 enter, 채팅 입력한 유저의 닉네임이 문자열로 주어짐
# 첫 번째 주어지는 문자열은 무조건 enter
# 중복 찾기 다음 엔터까지

from itertools import permutations
import sys

def read_input():
    try:
        # Try to read from a file
        with open('input.txt', 'r') as f:
            data = f.read().split()
    except FileNotFoundError:
        # If the file is not found, read from standard input
        data = sys.stdin.read().split()
    return data
data = read_input()

N = int(data[0])
chatting_record = list(map(str, data[1:1+N]))
general_chatting_list = {}
check_general_chatting_list = []

for i in range(0, len(chatting_record)):
  if chatting_record[i] != 'ENTER':
    #"ENTER"가 아닌 일반채팅 기록 key, value로 담기
    general_chatting_list[i] = chatting_record[i]

keys = list(general_chatting_list.keys())

for j in range(len(keys)-1):
  if keys[j] + 1 == keys[j+1]:
    if keys[j] in general_chatting_list:
      #딕셔너리 key는 직접 접근을 해줘야 value를 찾을 수 있음
      check_general_chatting_list.append(general_chatting_list[keys[j]])

if check_general_chatting_list:
  count_chatting = len(list(permutations(check_general_chatting_list)))
else:
  count_chatting = 0
  
print(count_chatting)