오픈채팅방 (Java)

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

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

 

프로그래머스 플랫폼에 올라와있는 2019 KAKAO BLIND RECRUITMENT "오픈채팅방" 문제 풀이이다.

관리자 메시지는 들어갈 때, 나갈 때만 기록하면 되지만 사용자가 이름을 변경하면 지난 이력도 바뀌어야 하므로 마지막으로 바뀐 이름을 알아야 한다.

 

 

첫 번째 방법 (시간 초과)

첫 번째로 생각했던 방법은 아래와 같다.

이력을 한 줄씩 읽으면서 들어갈 때 또는 나갈 때면 맨 끝에서부터 이력을 읽으면서 마지막으로 바뀐 이름을 조회해오도록 코드를 짰다.

뒤에서부터 읽으면서 바뀐 이름을 찾으면 반복 횟수가 적어 시간이 더 적게 들 줄 알았지만 데이터에 따라 오래 걸리는 경우가 있어서 시간 초과로 통과하지 못했다.

 

import java.util.*;

class Solution {
    public String[] solution(String[] records) {
        List<String> result = new ArrayList<>();
        for (String record : records) {
            if (record.startsWith("Enter") || record.startsWith("Leave")) {
                String[] splitedRecord = record.split(" ");
                String name = getLastName(records, splitedRecord[1]);
                if (record.startsWith("Enter")) {
                    result.add(name + "님이 들어왔습니다.");
                } else {
                    result.add(name + "님이 나갔습니다.");
                }
                
            }
        }
        return result.toArray(new String[result.size()]);
    }
    
    public String getLastName(String[] records, String userId) {
        for (int i = records.length - 1; i >= 0; i--) {
            String record = records[i];
            if (record.startsWith("Enter") || record.startsWith("Change")) {
                String[] splitedRecord = record.split(" ");
                if (splitedRecord[1].equals(userId)) {
                    return splitedRecord[2];
                }
            }
        }
        return "";
    }
}

 

 

두 번째 방법

마지막으로 바뀐 이름을 이력에서 매번 조회해오는 것이 비효율적인 것 같아 유저 아이디를 키로하는 맵에 저장하도록 변경했다.

유저 아이디별 최종 변경된 이름을 저장하는 반복문과 이력을 기록하는 반복문을 따로 돌려 총 이력을 2번 확인하기 때문에 좋지 않은 방법이긴 하지만 통과했다.

 

import java.util.*;

class Solution {
    public String[] solution(String[] records) {
        List<String> answer = new ArrayList<>();    // 결과 문자열 리스트
        Map<String, String> map = new HashMap<>();  // 최종 변경된 id와 name의 쌍
        
        // 입력된 변경 기록 중 Enter, Change 시마다 유저아이디 별 이름을 저장함
        for (String record : records) {
            if (!record.startsWith("Leave")) {
                String[] info = record.split(" ");
                map.put(info[1], info[2]);
            }
        }
        
        // 사용자가 들어오거나 나간 기록을 메시지로 변경
        // 사용자 이름은 최종 정보를 저장한 맵에서 가져옴
        for (String record : records) {
            if (!record.startsWith("Change")) {
                String[] info = record.split(" ");
                if (record.startsWith("Enter")) {
                    answer.add(map.get(info[1]) + "님이 들어왔습니다.");
                } else if (record.startsWith("Leave")) {
                    answer.add(map.get(info[1]) + "님이 나갔습니다.");
                }
            }
        }
        
        return answer.toArray(new String[answer.size()]);
    }
}

 

 

세 번째 방법

맵을 사용하니까 확장성, 가독성 측면에서 한계가 있어서 Dto를 사용하도록 변경해보았다.

또한, 문자열 포맷을 사용해서 변경이 쉽도록 하며 유동적인 문자열 부분을 맵에서 조회해오는 이름으로 치환하여 두 번째 방법과 로직은 비슷하다.

 

import java.util.*;

class Solution {
    private static final String ENTER_FORMAT = "%s님이 들어왔습니다.";
    private static final String LEAVE_FORMAT = "%s님이 나갔습니다.";
    
    private class User {
        public String userId;
        public String name;

        public User(String userId, String name) {
            this.userId = userId;
            this.name = name;
        }

    }

    private class Command {
        public String command;
        public String name;

        public Command(String command, String name) {
            this.command = command;
            this.name = name;
        }
    }

    public String[] solution(String[] records) {
        List<Command> commandList = new ArrayList<>();
        Map<String, User> userMap = new HashMap<>();

        for (String record : records) {
            String[] split = record.split(" ");
            String command = split[0];
            String userId = split[1];
            String name = null;

            switch(command) {
                case "Enter":   // 들어갈 때 유저아이디별로 사용자 정보(유저아이디, 이름) 업데이트 + 이력
                    name = split[2];
                    userMap.put(userId, new User(userId, name));
                    commandList.add(new Command(command, userId));
                    break;
                case "Leave":   // 나갈 때는 이력만
                    commandList.add(new Command(command, userId));
                    break;
                case "Change":  // 변경할 때는 기존 맵에 저장된 정보 변경
                    name = split[2];
                    userMap.get(userId).name = name;
                    break;
            }
        }

        return commandList.stream()
                // 저장된 메시지의 형식을 변경함(Enter -> ~들어왔습니다.)
                // 출력할 이름은 맵에 저장된 값으로 변경함
                .map(msg -> String.format(msg.command.equals("Enter") ? ENTER_FORMAT : LEAVE_FORMAT, userMap.get(msg.name).name))
                .toArray(ary -> new String[commandList.size()]);
    }
}