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

import java.util.*;
 
class Solution {
    private static final String ENTER_FORMAT = "%s님이 들어왔습니다.";
    private static final String LEAVE_FORMAT = "%s님이 나갔습니다.";
    
    public String[] solution(String[] records) {
        String[] answer = {};
        
        // <userId, nickName>
        Map<String, String> userNickMap = new HashMap<>();
        List<Trace> traceList = new ArrayList<>();
        
        for(String record : records){
            String[] split = record.split(" ");
            String cmd = split[0];
            String userId = split[1];
            String nickName = null;
            
            switch(cmd){
                case "Enter":
                    traceList.add(new Trace(userId, "Enter"));
                    nickName = split[2];
                    userNickMap.put(userId, nickName);
                    
                    // System.out.println(nickName);
                    break;
                case "Leave":
                    traceList.add(new Trace(userId, "Leave"));
                    // System.out.println(nickName);
                    break;
                case "Change":
                    nickName = split[2];
                    userNickMap.put(userId, nickName);
                    // System.out.println("캐릭캐릭ㅊ에인지");
                    break;
            }
        }
        
//         traceList.stream().forEach(t -> System.out.println(t.userId+" "+t.trace));
//         System.out.println();
//         traceList.stream().map(t->String.format( (t.trace.equals("Enter"))?ENTER_FORMAT:LEAVE_FORMAT, userNickMap.get(t.userId))).forEach(System.out::println);
            
        
        return traceList.stream().map(t->String.format( (t.trace.equals("Enter"))?ENTER_FORMAT:LEAVE_FORMAT, userNickMap.get(t.userId))).toArray(String[]::new);
    }
    
    public class Trace{
        String userId;
        String trace;
        
        Trace(String userId, String trace){
            this.userId = userId;
            this.trace = trace;
        }
    }
}

 

 

 

 

해쉬맵을 이용해 id에 맞는 nickname을 저장한다

 

첫 for문에는 해쉬맵에 id와 nickname을 매칭시켜준다

 

이 때 문자열 배열 answer의 길이를 넣기 위해 cnt값을 카운팅 시켜줌

 

다음 for문에는 string배열을 정의해주는데

 

들어온 경우와 나간 경우에만 이벤트가 발생하므로

 

그 경우에 맞춰서 앞서 정의한 해쉬맵을 활용하여

 

들어올 때와 나갈때를 구분하여 answer를 완성시켜준다

 

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46

import java.util.*; class Solution { public String[] solution(String[] record) { String id,nick,state; int cnt=0,index=0; Map<String,String> hm = new HashMap<>(); for (String str : record) { StringTokenizer tokenizer = new StringTokenizer(str); state=tokenizer.nextToken(); if(state.equals("Enter") || state.equals("Leave")) { cnt++; } if(state.equals("Enter") || state.equals("Change")) { id=tokenizer.nextToken(); nick=tokenizer.nextToken(); hm.put(id, nick); } } String[] answer = new String[cnt]; for (String str : record) { StringTokenizer tokenizer = new StringTokenizer(str); state=tokenizer.nextToken(); if(state.equals("Enter")) { id=tokenizer.nextToken(); nick=hm.get(id); answer[index++]=nick+"님이 들어왔습니다."; } else if (state.equals("Leave") ) { id=tokenizer.nextToken(); nick=hm.get(id); answer[index++]=nick+"님이 나갔습니다."; } } return answer; } }