[Coding Test Practice] 오픈채팅방 파이썬

🥯 링크

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

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

 

🥯 접근

고유한 [user_id]에 마지막 [user_nick]을 대입해야 한다.

[user_id]와 [user_nick]을 매치해줄 수 있는 딕셔너리(Dictionary)를 이용하자.

 

🥯 코드

def solution(record):
    answer = []
    
    # user_id : 들락날락하는 user의 id를 순서대로 담을 배열
    user_id = []
    
    # users : user_id 와 user_nick 을 매치시켜줄 딕셔너리
    users = {}
    
    for r in record:
        command = r.split() # 결과형식 예시 : ["Enter", "user_id", "user_nick"]
        # elc : Enter Leave Change 셋 중 하나
        elc = command[0]
        # uid : 현재 user_id
        uid = command[1]
        
        if elc == "Enter":
        	# users 딕셔너리에 uid : "user_nick" 매치시켜주기
            users[uid] = command[2]
            # 들락날락하는 user의 id를 순서대로 담기
            user_id.append(uid)
            # 정답 배열에 Enter를 뜻하는 "E" 담아놓기
            answer.append("E")
            
        elif elc == "Leave":
        	# 들락날락하는 user의 id를 순서대로 담기
            user_id.append(uid)
            # 정답 배열에 Leave를 뜻하는 "L" 담아놓기
            answer.append("L")
            
        else: # elc == "Change":
        	# users 딕셔너리에 uid : 바뀐 "user_nick" 매치시켜주기
            users[uid] = command[2]
            
    # 들락날락한 횟수동안
    for i in range(len(answer)):
    	# user_id 순서대로 users 딕셔너리에 매칭되는 user_nick 받아오기
        user_nick = users[user_id[i]]
        
        if answer[i] == "E":
            answer[i] = user_nick + "님이 들어왔습니다."
        else:
            answer[i] = user_nick + "님이 나갔습니다."
            
    return answer

 

🥯 후기

딕셔너리 활용은 필수적이다..

딕셔너리는 다음과 같이 형식에 제한을 갖지 않으며

>>> a[3] = [1,2,3]
>>> a
{1: 'a', 2: 'b', 'name': 'pey', 3: [1, 2, 3]}

 

삭제하는 법도 간단하다.

>>> del a[1]
>>> a
{2: 'b', 'name': 'pey', 3: [1, 2, 3]}

 

주의할 점은,

딕셔너리에서 Key는 고유한 값이므로 중복되는 Key 값을 설정해 놓으면 하나를 제외한 나머지 것들이 모두 무시된다는 점을 주의해야 한다. 다음 예에서 볼 수 있듯이 동일한 Key가 2개 존재할 경우 1:'a' 쌍이 무시된다.

>>> a = {1:'a', 1:'b'}
>>> a
{1: 'b'}

 

오늘은 간단히 정의에 대해서만 알고, 내일 딕셔너리에 관한 함수들도 공부하자.