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

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

 

소요시간 : 18분

 

간단히 map 사용할 줄 안다면 할 수 있는 문제..

 

record를 모두 돌면서

enter 와 change 때 map.put() 으로 id 와 name 넣어주면

자동으로 생성되거나 변경된다.

이 때 최종 출력은 enter와 leave 만 나오므로 횟수를 count하고

그 count 수 만큼 String 배열 크기 잡는다.

한 번 더 record 를 돌면서

enter, leave 명령에 따른 다른 메세지를 출력하며 그 때 이름은 map의 id 를 이용하여 뽑아낸다.

 

import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;

class Solution {
    public String[] solution(String[] record) throws Exception {
		Map<String, String> userMap = new HashMap<>();
		String[] arr;
		int cnt = 0;
		for (int i = 0; i < record.length; i++) {
			arr = record[i].split(" ");
			if(arr[0].equals("Enter")) {
				cnt++;
				userMap.put(arr[1], arr[2]);
			}
			else if(arr[0].equals("Change")) {
				userMap.put(arr[1], arr[2]);
			}
			else {
				cnt++;
			}
		}

		String[] answer = new String[cnt];
		int idx = 0;
		for (int i = 0; i < record.length; i++) {
			arr = record[i].split(" ");
			if(arr[0].equals("Enter")) {
				answer[idx++] = userMap.get(arr[1]) + "님이 들어왔습니다.";
			}
			else if(arr[0].equals("Leave")) {
				answer[idx++] = userMap.get(arr[1]) + "님이 나갔습니다.";
			}
		}
		return answer;
    }
}