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

문제

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

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

 

풀이

 

import java.util.ArrayList;
import java.util.HashMap;

class Solution {

    private HashMap<String, String> map = new HashMap<>();

    public String[] solution(String[] records) {
        ArrayList<String> answerList = new ArrayList<>();
        String message;

        for (String record : records) {
            message = process(record);
            if (message != null) {
                answerList.add(message);
            }
        }

        //set answer
        String[] answer = new String[answerList.size()];
        int idx = 0;
        for (String s : answerList) {
            String id = s.substring(0, s.indexOf('님'));
            answer[idx] = s.replace(id, map.get(id));
            ++idx;
        }
        return answer;
    }

    private String process(String r) {
        String[] temp = r.split(" ");

        String status = temp[0];
        String id = temp[1];
        if (temp.length == 3) {
            String name = temp[2];
            map.put(id, name);
        }

        if ("Enter".equals(status))
            return id + "님이 들어왔습니다.";
        else if ("Leave".equals(status))
            return id + "님이 나갔습니다.";
        return null;
    }

}