[프로그래머스] Lv2. 오픈채팅방

문제 링크: https://school.programmers.co.kr/learn/courses/30/lessons/42888

프로그래머스

코드

def solution(record):
    dict = {}
    arr = []
    for i in record:
        tmp = i.split()
        if len(tmp) == 3:
            dict[tmp[1]] = tmp[2]
    for i in record:
        tmp = i.split()
        if tmp[0] == 'Enter':
            msg = dict[tmp[1]]+"님이 들어왔습니다."
            arr.append(msg)
        elif tmp[0] == 'Leave':
            msg = dict[tmp[1]]+"님이 나갔습니다."
            arr.append(msg)
    return arr

풀이

• 첫 for문으로 모든 메시지를 돌아서, 딕셔너리에 아이디: 이름으로 저장한다. 이 때, Leave 명령어를 제외한 메시지는 모두 split하면 원소가 3개가 나오기 때문에, 원소가 3개일때에만 딕셔너리를 갱신하도록 했다.
• 이 문제는 들어온 사람이 이름을 바꾸면 바꾸기 전에 들어왔던 이름도 바뀌어서 출력되어야 하는데, 이미 첫 for문에서 Change 메시지에 따른 변화를 딕셔너리에 적용해놓았기때문에, Enter와 Leave 명령어만 확인해서 출력하면 된다.