[JAVA/코딩테스트] 문자열 압축

문제

1. 내가 푼 풀이

import java.util.*;
import java.io.*;

public class Main {
	
	public String solution(String input) {
		// 결과 문자열
        String result = "";
		// 같은 문자가 몇개인지 저장하는 카운터
		int count = 1;
		// 첫번째 문자를 결과값에 저장
		result = result + input.charAt(0);
		
        // 두번째 문자부터 반복문
		for(int i=1; i<input.length(); i++) {
			// 이전 문자와 지금 문자가 같을경우
			if(input.charAt(i-1) == input.charAt(i)) {
				// 카운터 ++
                count++;
                // 만약 문자 인덱스가 마지막일경우 카운터를 결과에 입력
				if(i==input.length()-1) {
					result = result + Integer.toString(count);
				}
			// 이전문자와 다를경우
            }else {
            	// 카운터가 1이상일 경우
            	if(count > 1) {
					result = result + Integer.toString(count) + input.charAt(i);
				// 카운터가 1일 경우
                }else {
					result = result + input.charAt(i);
				}
                // 카운터 초기화
				count = 1;
			}
				
		}
		
		return result;
	}

	public static void main(String args[]) {
        Main main = new Main();
		// 문자를 입력받는다
		Scanner in= new Scanner(System.in);
        // 첫줄 문자열
		String input = in.next();
		System.out.print(main.solution(input)); 
	}

}

2. 문자열 마지막에 공백을 넣어줘서 코드를 깔끔하게 구현

import java.util.*;
import java.io.*;

public class Main {
	
	public String solution(String input) {
		String result = "";
		
		input = input + " ";
		int cnt = 1;
		for(int i=0; i<input.length()-1; i++) {
			if(input.charAt(i) == input.charAt(i+1)) {
				cnt ++;
			}else {
				result += input.charAt(i);
				if(cnt > 1) {
					result += String.valueOf(cnt);
					cnt = 1;
				}
			}
		}
		
		return result;
	}

	public static void main(String args[]) {
        Main main = new Main();
		// 문자를 입력받는다
		Scanner in= new Scanner(System.in);
        // 첫줄 문자열
		String input = in.next();
		System.out.print(main.solution(input)); 
	}

}