[JAVA/구현] 프로그래머스 오픈채팅방

1. 문제

https://programmers.co.kr/learn/courses/30/lessons/42888?language=java 

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

 

2. 풀이

• 입력 받은 명령의 내용(입장, 퇴장)을 유저의 아이디와 함께 저장한다.
• 아이디와 이름을 해시맵에 저장한다. 아이디 변경을 반복하게 되는 경우 뒤에 들어온 value가 덮여 저장된다.
• 입력을 모두 받은 후, replace를 사용해서 유저의 아이디를 이름으로 교체한다.

 

3. 코드

import java.util.*;

class Solution {
    public String[] solution(String[] record) {
        
        HashMap<String, String> hMap = new HashMap<String, String>(); // id, name
        String enterMsg = "님이 들어왔습니다.";
        String leaveMsg = "님이 나갔습니다.";
        
        ArrayList<String> res = new ArrayList<String>();
        int idx = 0;
        for(int i = 0; i < record.length; i++){
            String[] ss = record[i].split(" ");
            
            if(ss[0].equals("Enter")){ // 입장
                hMap.put(ss[1], ss[2]); 
                res.add(ss[1] + enterMsg);
            }else if(ss[0].equals("Leave")){ // 퇴장
                res.add(ss[1] + leaveMsg);
            }else if(ss[0].equals("Change")){ // 닉네임 변경
                hMap.put(ss[1], ss[2]);
            }
            
        }
        
        String[] answer = new String[res.size()];
        
        for(int i = 0; i < res.size(); i++){
            String str = res.get(i);
            int endOfId = str.indexOf("님");
            String userId = str.substring(0, endOfId);
            
            answer[i] = str.replace(userId, hMap.get(userId)); // 닉네임 설정
            
        }
        
        
        return answer;
    }
}