2019 카카오 신입 오픈채팅방 level 2 - java

 

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

닉네임을 변경하면 이전에 입장했던 유저의 닉네임도 변경해주어야 한다.

그러므로 먼저 Map<userId, nickname> 을 설정해줬다.

그 뒤, record를 순회하면서 StringBuilder에 append해주었다.

 

import java.util.*;

class Solution {
    public String[] solution(String[] record) {
        String[] answer;
        List<String> chat = new ArrayList<>();
        Map<String, String> map = new LinkedHashMap<>();
        String enter = "님이 들어왔습니다.";
        String leave = "님이 나갔습니다.";
        int length = record.length;
        for(int i = 0; i<length;i++) {
            StringTokenizer st = new StringTokenizer(record[i]);
            String action = st.nextToken();
            if(action.equals("Leave")) {
                continue;
            }
            String userId = st.nextToken();
            String nickname = st.nextToken();
            map.put(userId, nickname);
        }
        for(int i = 0;i<length;i++) {
            StringTokenizer st = new StringTokenizer(record[i]);
            String action = st.nextToken();
            String userId = st.nextToken();
            if(action.equals("Change")) {
                continue;
            }
            StringBuilder sb = new StringBuilder();
            sb.append(map.get(userId));
            sb.append(action.equals("Enter")?enter:leave);
            chat.add(sb.toString());
        }
        answer = new String[chat.size()];
        for(int i = 0 ;i<chat.size();i++) {
            answer[i] = chat.get(i);
        }
        return answer;
    }
}