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

매우 간단한데,, 너무 복잡하게 풀어서 시간초과. 다시 풀었다.

 

- 닉네임은 계속 바뀌기 때문에 굳이 Enter, Change에서 기존 결과값을 계속 변경해줄 필요 없다.

- 다만 닉네임은 key-value 형태로 map에 저장,
- 마지막에 String[] answer로 옮기는 과정에서 최종 value를 기준으로 String을 만들어 넣어주기만 하면 된다.

 

 

+ 아래 코드로, arrayList to array가 가능하다.

answer = arrayList.toArray(new String[arrayList.size()]);

import org.junit.Test;

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

public class Solution {

    class UserInfo {
        String info;
        String id;

        public UserInfo(String info, String id) {
            this.info = info;
            this.id = id;
        }
    }

    public String[] solution(String[] record) {
        ArrayList<UserInfo> arrayList = new ArrayList<>();
        String[] answer;
        HashMap<String, String> hm = new HashMap<>();

        for (String x : record) {
            String[] split = x.split(" ");

            if (split[0].equals("Enter")) {
                hm.put(split[1], split[2]);
                arrayList.add(new UserInfo("님이 들어왔습니다.", split[1]));
                continue;
            }

            if (split[0].equals("Leave")) {
                String tmp = hm.get(split[1]);
                arrayList.add(new UserInfo("님이 나갔습니다.", split[1]));
                continue;
            }
            if (split[0].equals("Change")) {
                //이전 기록 id의 value를 갱신
                hm.put(split[1], split[2]);
            }
        }

        answer = new String[arrayList.size()];
        int i = 0;
        for(UserInfo usr : arrayList) {
            answer[i] = hm.get(usr.id) + usr.info;
            i++;
        }
        return answer;
    }

    @Test
    public void 정답() {
//        String[] rcd = {"Enter uid1234 Muzi", "Enter uid4567 Prodo", "Leave uid1234"};
//        String[] rcd = {"Enter uid1234 Muzi", "Enter uid4567 Prodo", "Leave uid1234", "Enter uid1234 prodo", "Change uid4567 Ryan"};
//        String[] rcd = {"Enter uid1234 Muzi", "Enter lee Prodo", "Enter uid3423 Prodo", "Change lee prodo_jh", "Leave lee", "Enter lee JH"};
        String[] rcd = {"Enter uid1234 Muzi", "Enter lee Prodo", "Enter uid3423 Prodo", "Change lee prodo_jh", "Leave lee"};
        this.solution(rcd);
    }
}