오픈채팅방 level2

문제를 읽은 후 생각

- 파이썬 dict를 이용하자

- name dict를 만든 후 for loop을 돌며 answer array를 채워가자

 

코드

def solution(record):
    answer = []
    enter = '님이 들어왔습니다.'
    leave = '님이 나갔습니다.'
    name_dict = dict()
    for info in record:
        info = info.split()
        if info[0] == 'Enter':
            name_dict[info[1]] = info[2]
        elif info[0] == 'Change':
            name_dict[info[1]] = info[2]
    
    for info in record:
        info = info.split()
        if info[0] == 'Enter':
            answer.append(name_dict[info[1]] + enter)
        if info[0] == 'Leave':
            answer.append(name_dict[info[1]] + leave)
        
    return answer

 

고찰

- 생각한대로 풀렸다.

- for loop을 한 번만 쓰고 해결할 수는 없을까?