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

문제

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

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

 

 

풀이

HashMap을 이용해 "유저 아이디"를 키로 "닉네임"을 값으로 저장한다.

HashMap의 put 메서드를 이용해 "Enter" 나 "Change" 명령에서 "유저 아이디"에 매칭되는 "닉네임"을 변경한다.

닉네임 대신에 유저 아이디를 리스트에 저장해서 마지막에 수정한 닉네임을 설정할 수 있게 한다. 

 

 

코드

java

import java.util.ArrayList;
import java.util.HashMap;
import java.util.StringTokenizer;
 
class Solution {
    public String[] solution(String[] record) {
 
        HashMap<String, String> hashMap = new HashMap<>();
        ArrayList<String> uidList = new ArrayList<>();
        ArrayList<String> enterList = new ArrayList<>();
 
        for (String str : record) {
            StringTokenizer stringTokenizer = new StringTokenizer(str);
            String cmd = stringTokenizer.nextToken();
            String uid = stringTokenizer.nextToken();
            String nickname = null;
 
            if (!cmd.equals("Leave"))
                nickname = stringTokenizer.nextToken();
 
            switch (cmd) {
                case "Enter":
                    hashMap.put(uid, nickname);
                    uidList.add(uid);
                    enterList.add("님이 들어왔습니다.");
                    break;
                case "Change":
                    hashMap.put(uid, nickname);
                    break;
                case "Leave":
                    uidList.add(uid);
                    enterList.add("님이 나갔습니다.");
                    break;
                default:
                    break;
            }
        }
 
        ArrayList<String> resultList = new ArrayList<>();
 
        for (int i=0; i<uidList.size(); i++) {
            resultList.add(hashMap.get(uidList.get(i)) + enterList.get(i));
        }
 
        String[] answer = resultList.toArray(new String[resultList.size()]);
        return answer;
    }
}

 

python

def solution(record):
    uid_nickname_dic = {}
 
    for r in record:
        chat = r.split()
        if chat[0] == "Enter" or chat[0] == "Change":
            uid_nickname_dic[chat[1]] = chat[2]
        
    answer = []
    for r in record:
        chat = r.split()
        if chat[0] == "Enter":
            answer.append(f"{uid_nickname_dic[chat[1]]}님이 들어왔습니다.")
        if chat[0] == "Leave":
            answer.append(f"{uid_nickname_dic[chat[1]]}님이 나갔습니다.")
            
    return answer