programmers_[3차] 압축_java

https://school.programmers.co.kr/learn/courses/30/lessons/17684

프로그래머스

import java.util.*;

class Solution {
    public int[] solution(String msg) {
        int[] answer;
        List<Integer> list = new ArrayList<>();
        
        // 사전
        Map<String, Integer> dict = new HashMap<>();
        
        // A - Z 초기화
        int idx= 1;
        for(int i = 0; i<26; i++){ 
            dict.put(Character.toString('A' + i), idx++);
        }
        
        int msgLength = msg.length();
        
        
        StringBuilder sb = new StringBuilder();
        
        for(int i = 0; i<msgLength; i++){         
            // 현재 입력 w가 사전에 있는 경우
            if(dict.containsKey(sb.toString() + Character.toString(msg.charAt(i)))){
                sb.append(Character.toString(msg.charAt(i)));
                continue;
            }
            // 현재 입력 w가 사전에 없는 경우
            else{
                list.add(dict.get(sb.toString()));
                dict.put(sb.toString() + Character.toString(msg.charAt(i)), idx++);                
                sb.setLength(0);
                sb.append(Character.toString(msg.charAt(i)));
            }    
        }
        
        // 남은 것 처리
        list.add(dict.get(sb.toString()));
        
        // list to array;
        int listSize = list.size();
        answer = new int[listSize];
        int tmp = 0;
        for(int i : list){
            answer[tmp++] = i;
        }
        return answer;
    }
}