오픈채팅방

코딩테스트 연습 - 오픈채팅방 | 프로그래머스 (programmers.co.kr)

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

풀이)

채팅방에 들어오고 나가는것을 로그를 기록해야 하는 문제인데 설정한 닉네임으로 로그를 남긴다. 중간에 사용자의 닉네임이 바뀐다면 해당 사용자에 의해 찍혔던 이전까지의 기록에도 바뀐 닉네임으로 기록되어야 한다.

문제를 해결하기 위해 우선 닉네임이 아닌 사용자의 아이디를 이용해 다음과 같이 메시지를 리스트에 저장한다.

['user1234', '님이 들어왔습니다.']

['user1234', '님이 나갔습니다.']

사용자가 채팅방에 입장하거나 닉네임을 변경할 때는 딕셔너리를 변경해준다.

이렇게 하면 userID와 출입메시지가 담긴 리스트와 userID:nickName이 담긴 딕셔너리가 생성된다.

 

이제 출입메시지가 담긴 리스트를 탐색하면서 userID에 맞는 nickName을 연결시켜 answer 리스트에 추가해준다.

코드)

def solution(record):
    answer = []
    userId = {}
    user = []
    for userData in record:
        u = userData.split(' ')
        a = []
        if u[0] == 'Enter':
            a.append(u[1]) # 아이디 추가
            a.append('님이 들어왔습니다.')
            userId[u[1]] = u[2] # 닉네임 등록
            answer.append(a)
            
        elif u[0] == 'Leave':
            a.append(u[1]) 
            a.append('님이 나갔습니다.')
            answer.append(a)    
        else:
            userId[u[1]] = u[2] # 닉네임 변경
        
    result = []
    for lst in answer: # ['uid1234',님이 들어왔습니다]
        result.append(userId[lst[0]] + lst[1])  #userId[lst[0]]  == Prodo

    return result