[프로그래머스] 오픈채팅방

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

프로그래머스

---

처음 제출한 코드 - 31개 테케에서 런타임 에러

def solution(record):
    answer = []
    user = dict()
    msg = []
    for rec in record:
        temp = rec.split()
        if temp[0] == "Leave":
            msg.append(temp[1]+"님이 나갔습니다.")
            continue
        elif temp[0] == "Enter":
            msg.append(temp[1]+"님이 들어왔습니다.")
        user[temp[1]] = temp[2]
        
    for m in msg:
        uid = m[:7]
        answer.append(m.replace(uid,user[uid]))
    return answer

문제 조건을 다시 읽어보니 user id 길이가 uidxxxx 총 7글자로 고정된 것이 아니라, 1 이상 10 이하의 문자열이라고 했다.

즉 밑에서 세번째 line에서 uid를 긁어오는 부분이 틀렸다.

 

수정한 코드 (정답)

def solution(record):
    answer = []
    user = dict()
    msg = []
    for rec in record:
        temp = rec.split()
        if temp[0] == "Leave":
            msg.append(temp[1]+"님이 나갔습니다.")
            continue
        elif temp[0] == "Enter":
            msg.append(temp[1]+"님이 들어왔습니다.")
        user[temp[1]] = temp[2]
        
    for m in msg:
        uid = m.split()[0][:-2]
        answer.append(m.replace(uid,user[uid]))
    return answer