[실전 문제] 문자열 압축

1. 문제

문자열의 길이가 굉장히 길 경우, 이를 압축하여 짧게 만들어야 할 때가 종종 있다. 이 문제에서는 문자열이 주어졌을 때, 같은 알파벳이 연속된 부분 문자열을 압축하여 결과를 출력하는 프로그램을 작성한다. 예를 들어, 문자열이 AAABBBBBCCCCDDDDEFFF 라고 하자. 이 문자열을 압축하면, 연속으로 같은 문자가 나오는 부분에, 그 문자가 몇번 나왔는지를 적어줌으로써 압축한다. 즉, 이 문자열은 3A5B4C4DE3F 로 압축된다. E는 1개밖에 없기 때문에 따로 1을 적어주지 않는다.

---

입력

첫 번째 줄에 압축하고자 하는 문자열이 주어진다. 문자열의 길이는 1000보다 작다. 문자열에 구성된 알파벳은 대문자다. 

출력

문자열을 압축한 결과를 출력한다.

예제 입력

AAABBBBBCCCCDDDDEFFF

예제 출력

3A5B4C4DE3F

 

 

2. 풀이

이 문제의 경우, 마지막 요소에 대한 값을 반환하지 못하는 부분이 있다. 따라서 주어진 입력 뒤에 공백을 추가하여 마지막 요소까지 처리할 수 있도록 하면 해결할 수 있다.

import java.io.*;

public class Main {

    private static final BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    private static final BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

    public static void main(String[] args) throws IOException {

        String str = br.readLine();

        str += ' ';
        int count = 1;
        for(int i = 0; i < str.length()-1; i++) {
            if(str.charAt(i) == str.charAt(i+1)) count++;
            else {
                if(count > 1) bw.write(count + "" + str.charAt(i));
                else bw.write(str.charAt(i) + "");
                count = 1;
            }
        }
        br.close();
        bw.flush();
        bw.close();
    }

}