[JAVA] 백준 1343 - 폴리오미노

https://www.acmicpc.net/problem/1343

 

입력을 받은 후 여러개의 마침표로 split된 배열 하나, 여러개의 X로 split된 배열 하나를 만들고

마침표로 split된 배열에 대해 만약 length가 홀수면 -1 반환 후 함수를 종료하고

짝수라면 최대한 AAAA를 채운 뒤 나머지에 대해 BB를 채운 문자열을 생성해 배열에 저장했다.

그 이후 입력 문자열의 각 character를 읽어 X라면 AB로 변환한 배열에서 순번에 맞는 문자열을 가져와 붙이고 

반복문의 index를 가져온 문자열의 길이 -1 만큼 늘리고, 마침표라면 마침표를 모아놓은 배열에서 가져와 붙여서

정답 문자열을 만들었다.

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String s = br.readLine();
        String[] t = s.split("[.]+");
        String[] dot = s.split("X+");
        for (int j = 0; j<t.length; j++) {
            String string = t[j];
            StringBuilder sb = new StringBuilder();
            if(string.length()%2==1){
                System.out.println(-1);
                return;
            }else{
                int count4 = string.length()/4;
                for (int i = 0; i < count4; i++) {
                    sb.append("AAAA");
                }
                int left = (string.length() - (count4*4))/2;
                for (int i = 0; i < left; i++) {
                    sb.append("BB");
                }
            }
            t[j] = sb.toString();
        }
        StringBuilder sb = new StringBuilder();
        int tidx = 0;
        int didx = 0;
        for (int i = 0; i < s.length(); i++) {
            if(s.charAt(i)=='X'){
                sb.append(t[tidx]);
                i+=t[tidx].length()-1;
                tidx++;
            }else{
                sb.append(dot[didx]);
                i+=dot[didx].length()-1;
                didx++;
            }
        }
        System.out.println(sb);

    }
}