[프로그래머스] 오픈채팅방 - 파이썬/python

문제

https://school.programmers.co.kr/learn/courses/30/lessons/42888

프로그래머스

오픈채팅방에 들어오고 나가거나, 닉네임을 변경한 기록을 담은 문자열 배열이 주어질 때, 모든 기록이 처리된 후, 최종적으로 방을 개설한 사람이 보게 되는 메시지를 반환하는 문제다.

 

제한사항

• record는 다음과 같은 문자열이 담긴 배열이며, 길이는 1 이상 100,000 이하이다.
• 다음은 record에 담긴 문자열에 대한 설명이다.
  • 모든 유저는 [유저 아이디]로 구분한다.
  • [유저 아이디] 사용자가 [닉네임]으로 채팅방에 입장 - "Enter [유저 아이디] [닉네임]" (ex. "Enter uid1234 Muzi")
  • [유저 아이디] 사용자가 채팅방에서 퇴장 - "Leave [유저 아이디]" (ex. "Leave uid1234")
  • [유저 아이디] 사용자가 닉네임을 [닉네임]으로 변경 - "Change [유저 아이디] [닉네임]" (ex. "Change uid1234 Muzi")
  • 첫 단어는 Enter, Leave, Change 중 하나이다.
  • 각 단어는 공백으로 구분되어 있으며, 알파벳 대문자, 소문자, 숫자로만 이루어져있다.
  • 유저 아이디와 닉네임은 알파벳 대문자, 소문자를 구별한다.
  • 유저 아이디와 닉네임의 길이는 1 이상 10 이하이다.
  • 채팅방에서 나간 유저가 닉네임을 변경하는 등 잘못 된 입력은 주어지지 않는다.

 

예시

record:
["Enter uid1234 Muzi", "Enter uid4567 Prodo","Leave uid1234","Enter uid1234 Prodo","Change uid4567 Ryan"]
result:
["Prodo님이 들어왔습니다.", "Ryan님이 들어왔습니다.", "Prodo님이 나갔습니다.", "Prodo님이 들어왔습니다."]

 

풀이

def solution(record):
    answer = []
    user_dict = {}
    actions = []
    
    # 1. 유저 정보를 갱신하며 이벤트를 기록한다.
    for i in record:
        action = i.split()
        if action[0] == 'Enter':
            user_dict[action[1]] = action[2]
            actions.append((action[0], action[1]))
        elif action[0] == 'Leave':
            actions.append((action[0], action[1]))
        elif action[0] == 'Change':
            user_dict[action[1]] = action[2]
    
    # 2. 이벤트를 기반으로 결과 메시지를 작성한다.
    for action, user_id in actions:
        if action == 'Enter':
            answer.append(f"{user_dict[user_id]}님이 들어왔습니다.")
        elif action == 'Leave':
            answer.append(f"{user_dict[user_id]}님이 나갔습니다.")
    
    return answer

• 유저 아이디와 닉네임을 매핑할 딕셔너리와 각 이벤트(Enter, Leave, Change)를 기록할 리스트 만든다.
• 주어진 문자열을 순회하며 유저 정보를 user_dict 딕셔너리에 갱신하고, 이벤트를 actions 리스트에 기록한다.
• actions 리스트를 순회하며 각 이벤트와 해당 유저 정보를 기반으로 메시지를 정답 리스트에 추가한다.