2018 KAKAO BLIND RECRUITMENT : [3차] 압축

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

코딩테스트 연습 - [3차] 압축

 

// 코드

import java.util.*;
 
class Solution {
    public int[] solution(String msg) {
        
        ArrayList<String> index = new ArrayList<>(Arrays.asList(new String[]{"A","B","C","D","E","F","G","H","I","J","K","L","M","N","O","P","Q","R","S","T","U","V"
                ,"W","X","Y","Z"}));
        
        if(msg.length() == 1) {
            int i = index.indexOf(msg);            
            return new int[]{ i+1 };
        }
        
        ArrayList<Integer> res = new ArrayList<>();
        String w = String.valueOf(msg.charAt(0));
        int idx = 1;
                
        while(idx <= msg.length()) {
            
            if(idx == msg.length()) {
                res.add((index.indexOf(w) + 1));
                break;
            }
            
            char c = msg.charAt(idx);
            
            String wc = w + c;
            
            if(index.contains(wc)) {
                w = wc;
                idx++;
                continue;
            }
            
            index.add(wc);
            res.add(index.indexOf(w) + 1);
            
            w = String.valueOf(c);
            idx++;
        }
        
        int[] answer = new int[res.size()];
        for(int i=0; i<answer.length; i++) answer[i] = res.get(i);
        return answer;
    }
}