[Programmers] Lv2. 오픈채팅방

문제

바로가기

코딩테스트 연습 - 오픈채팅방

 

풀이

• 먼저, record를 돌면서 들어오거나 변경되는 유저의 아이디와 닉네임을 딕셔너리에 저장하는 과정을 거침
• 그러면 변경된 이름도 적용되어 있기 때문에, 다시 배열을 돌면서 Enter와 Leave에 해당하는 메시지를 남길 수 있음

 

코드

def solution(record):
    answer=[]
    nickname = {}
    for i in [i.split(' ') for i in record]:
        if i[0] != 'Leave':
            nickname[i[1]] = i[2]
    for i in [i.split(' ') for i in record]:
        if i[0] == 'Enter':
            answer.append(nickname[i[1]] + '님이 들어왔습니다.')
        if i[0] == 'Leave':
            answer.append(nickname[i[1]] + '님이 나갔습니다.')
    return answer