57-xxubin04

[xxubin04]

🔗 문제 링크

프로그래머스 - 오픈채팅방

✔️ 소요된 시간

1시간 + a

✨ 수도 코드

1. 문제 이해

닉네임 변경방법에는 2가지가 존재한다.

• 채팅방을 나간 후, 새로운 닉네임으로 다시 들어감
• 채팅방에 속해있는 상태에서 닉네임 변경 -> 기존의 닉네임도 함께 변경된다.

• 중복 닉네임 허용

예를 들어보면, 다음과 같다.

1. Prodo님이 들어왔습니다. (uid1234, Muzi가 채팅방 들어옴)
2. Ryan님이 들어왔습니다. (uid4567, Prodo가 채팅방 들어옴)
3. Prodo님이 나갔습니다. (uid1234, Muzi가 닉네임을 Prodo로 바꾸고 난 후에 채팅방 나감)
4. Prodo님이 들어왔습니다. (uid1234, Muzi(Prodo)가 채팅방 들어옴)
5. (uid4567, Prodo가 Ryan으로 닉네임 바꿈)

즉, 가장 마지막에 정해진 닉네임으로 모든 채팅방 메시지가 변환된다.
그래서 유저 아이디를 key로 하여 (행동, 닉네임)을 저장한다. 그리고 가장 마지막에 저장된 (행동, 닉네임)에서 다음의 조건을 따져 각각의 유저 아이디에 하나의 닉네임만을 정해준다.

가장 마지막에 저장된 튜플에서 행동이 Enter거나 Change의 닉네임 = 확정 닉네임

하나의 닉네임을 각각 지정해주고 나서 Record에 나오는 순서대로 값을 반환해준다.

2. 코드 분석

    answer = []
    users = {}  # 사용자 아이디를 기준으로 (행동, 닉네임) 저장
    
    for r in record:  
        splitRecord = r.split()  # 공백을 기준으로 나눠줌 ("Enter", "uid1234", "Muzi")
        command = splitRecord[0]  # 행동
        userId = splitRecord[1]  # 사용자 아이디
        
        if command == "Leave":  # 행동이 Leave라면 닉네임이 주어지지 않으므로 따로 확인해준다.
            if userId in users:   # (행동, 닉네임) 튜플을 추가해준다.
                users[userId].append((command, ""))
            else:
                users[userId] = [(command, "")]
        else:
            if userId in users:
                users[userId].append((command, splitRecord[2]))
            else:
                users[userId] = [(command, splitRecord[2])]

record = ["Enter uid1234 Muzi", "Enter uid4567 Prodo","Leave uid1234","Enter uid1234 Prodo","Change uid4567 Ryan"]

users = ["uid1234" : [("Enter", "Muzi"), ("Leave", ""), ("Enter", "Prodo")], "uid4567" : [("Enter", "Prodo"), ("Change", "Ryan")]]]

위와 같이 저장된다.

    nicknames = {}

    for userId, action in users.items():  
        for command, name in reversed(action):  # users에 저장된 각각의 사용자 아이디의 value인 튜플
            if command == "Enter" or command == "Change":  # 만약 행동이 Enter나 Change라면 닉네임 확정
                nicknames[userId] = name
                break

튜플을 역순으로 순회하면서 행동이 Enter나 Change인 경우에는 그때의 닉네임으로 결정짓고 for문을 탈출한다. (최신의 닉네임으로 확정짓기 위함)

nicknames에 각각의 사용자 아이디에 따라 닉네임을 저장해준다.

def msg(act, name):
    if act == "Enter":
        return name+"님이 들어왔습니다."
    else:  # act == "Leave"
        return name+"님이 나갔습니다."

    for output in record:
        splitOutput = output.split()
        if splitOutput[0] != "Change":  # 행동이 Change가 아니라면,
            answer.append(msg(splitOutput[0], nicknames[splitOutput[1]]))

record 순서대로 answer를 반환해야 하므로 record의 각 문자열을 output으로 받아 공백을 기준으로 split해준다.
행동이 Change가 아니라면, msg함수를 호출해준다.

msg함수

행동이 Enter라면, return name+"님이 들어왔습니다."
행동이 Leave라면, return name+"님이 나갔습니다."

3. 전체 코드

 def msg(act, name):
    if act == "Enter":
        return name+"님이 들어왔습니다."
    else:  # act == "Leave"
        return name+"님이 나갔습니다."


def solution(record):
    answer = []
    users = {}

    for r in record:
        splitRecord = r.split()
        command = splitRecord[0]
        userId = splitRecord[1]

        if command == "Leave":
            if userId in users:
                users[userId].append((command, ""))
            else:
                users[userId] = [(command, "")]
        else:
            if userId in users:
                users[userId].append((command, splitRecord[2]))
            else:
                users[userId] = [(command, splitRecord[2])]

    nicknames = {}

    for userId, action in users.items():
        for command, name in reversed(action):
            if command == "Enter" or command == "Change":
                nicknames[userId] = name
                break

    for output in record:
        splitOutput = output.split()
        if splitOutput[0] != "Change":
            answer.append(msg(splitOutput[0], nicknames[splitOutput[1]]))

    return answer

📚 새롭게 알게된 내용

[9kyo-hwang]

우리에게 필요한 건 "최종적인 닉네임으로 유저가 들어오고 나간 기록"이잖아요? 즉 우리는

1. record를 한 번 순차적으로 훑어보며 각 유저 별 최종 닉네임을 기록
2. record를 한 번 더 순회해 Enter/Leave에 대해 [최종 닉네임]님이 들어왔/나갔습니다 문자열 만들기

이 2단계를 수행하면 됩니다. 그러면 코드가 단순해지죠.

# r.split()[0]: Enter/Leave/Change
# r.split()[1]: id
# r.split()[2]: nickname
id_to_nickname = {r.split()[1] : r.split()[2] for r in record if not r.startswith('Leave')}

우선 record를 차례대로 순회해 각 유저 별 최종적인 닉네임을 획득한다. Enter/Change에 의해 닉네임이 결정된다.

message = {'Enter':'님이 들어왔습니다.', 'Leave':'님이 나갔습니다.'}

문자열을 만들기위해 필요한 딕셔너리를 선언해준다.

# r.split()[1]: id -> id_to_nickname[r.split()[1]]: id에 대응되는 최종 닉네임
# r.split()[0]: Enter/Leave/Change -> message[r.split()[0]]: 각각에 대응되는 문자열
return [f'{id_to_nickname[r.split()[1]]}{message[r.split()[0]]}' for r in record if not r.startswith('Change')]

최종적으로 우리는 들어온/나간 문자열이 필요하므로 Enter, Leave에 대해 id에 대응되는 닉네임과 message를 이용해 문자열을 만들어 리스트에 저장한 뒤 반환한다.

전체 코드

def solution(record):
    id_to_nickname = {r.split()[1] : r.split()[2] for r in record if not r.startswith('Leave')}
    message = {'Enter':'님이 들어왔습니다.', 'Leave':'님이 나갔습니다.'}
    return [f'{id_to_nickname[r.split()[1]]}{message[r.split()[0]]}' for r in record if not r.startswith('Change')]

[mjj111]

예전에 풀었던 파이썬 코드입니다만, 이중 for문 필요없이
O(n) 안에 풀 수 있습니다..!

def solution(record):
    answer = []
    dic = {}
    
    for sentence in record:
        sentence_split = sentence.split()
        if len(sentence_split) == 3:
            dic[sentence_split[1]] = sentence_split[2]
            
    for sentence in record:
        sentence_split = sentence.split()
        
        if sentence_split[0] == 'Enter':
            answer.append(f'{dic[sentence_split[1]]}님이 들어왔습니다.' )
        elif sentence_split[0] == 'Leave':
            answer.append(f'{dic[sentence_split[1]]}님이 나갔습니다.')
            
    return(answer)