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

문제

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

프로그래머스

 

설계

파이썬

def solution(record):
    # split으로 빈칸을 구분해 list로 저장한다.
    arr, N = [], len(record)
    for i in range(N):
        arr.append(record[i].split())

    # dictionary를 사용해 변경된 id를 저장한다.
    name = {}
    for i in range(N):
        if len(arr[i]) == 2:
            continue
        name[arr[i][1]] = arr[i][2]

    # 채팅방 최종 메세지 저장 (Change는 추가하지 않는다)
    answer = []
    for i in range(N):
        if arr[i][0] == 'Enter':
            answer.append(f'{name[arr[i][1]]}님이 들어왔습니다.')
        elif arr[i][0] == 'Leave':
            answer.append(f'{name[arr[i][1]]}님이 나갔습니다.')

    return answer