[프로그래머스] 압축

문제

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

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

문제해결방법

문제에 나와 있는 LZW 압축 과정을 그대로 코딩해줘서 풀었습니다.

코드

package Programmers;
 
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
 
class Word {
    int index;
    String word;
    
    Word(int index, String word) {
        this.index = index;
        this.word = word;
    }
}
 
public class Solution_압축 {
 
    public static List<Word>[] words = new ArrayList[50];
    
    public static boolean startWith(String s, String word) {
        boolean check = true;
        if(s.length()<word.length()) check = false;
        else {
            for(int i=0; i<word.length(); i++) {
                if(s.charAt(i)!=word.charAt(i)) {
                    check = false;
                    break;
                }
            }
        }
        return check;
    }
    
    public static int[] solution(String msg) {
        List<Integer> indexList = new ArrayList<>();
        for(int i=0; i<words.length; i++) words[i] = new ArrayList<>();
        for(int i=1; i<=26; i++) words[1].add(new Word(i, (char)('A'+i-1)+""));
        int maxWordLength = 1;
        int nextIndex = 27;
        loop:while(msg.length()>0) {
            for(int i = maxWordLength; i>0; i--) {
                for(int j=0; j<words[i].size(); j++) {
                    if(startWith(msg, words[i].get(j).word)) {
                        String word = words[i].get(j).word;
                        int length = word.length();
                        indexList.add(words[i].get(j).index);
                        msg = msg.substring(length);
                        if(msg.length()!=0) {
                            words[length+1].add(new Word(nextIndex++, word+msg.charAt(0)));
                        }
                        if(maxWordLength<length+1) maxWordLength=length+1;
                        continue loop;
                    }
                }
            }
        }
        int[] answer = new int[indexList.size()];
        for(int i=0; i<answer.length; i++) answer[i] = indexList.get(i);
        return answer;
    }
    
    public static void main(String[] args) {
        String msg = "";
        System.out.println(Arrays.toString(solution(msg)));
    }
}

코드에 대한 피드백이나 더 좋은 아이디어는 언제나 환영입니다.