Baekjoon 1343 폴리오미노 JAVA

1343번: 폴리오미노

문제

민식이는 다음과 같은 폴리오미노 2개를 무한개만큼 가지고 있다. AAAA와 BB

이제 '.'와 'X'로 이루어진 보드판이 주어졌을 때, 민식이는 겹침없이 'X'를 모두 폴리오미노로 덮으려고 한다. 이때, '.'는 폴리오미노로 덮으면 안 된다.

폴리오미노로 모두 덮은 보드판을 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 보드판이 주어진다. 보드판의 크기는 최대 50이다.

출력

첫째 줄에 사전순으로 가장 앞서는 답을 출력한다. 만약 덮을 수 없으면 -1을 출력한다.

풀이

입력받는 문자열을 탐색하면서 X의 연속된 길이를 센다

이때 길이가 홀수이면 덮을 수 없기에 -1을 출력한다.

길이가 짝수이면 4로 나눈 몫만큼 AAAA로 덮고, 나머지는 2의 몫만큼 BB로 덮는다.

코드

import java.io.*;
 
public class Main {
	public static void main(String[] args) throws Exception{
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		String str = br.readLine();
		StringBuilder sb = new StringBuilder();
		int i=0,j=0, len=str.length();
		while(i<len) {
			if(str.charAt(i)=='X' ) {
				j=0;
				while(i+j<len && str.charAt(i+j)=='X')
					j++;
				if(j%2==1) {
					i=len;
					sb.setLength(0);
					sb.append("-1");
				}
				else {
					sb.append("AAAA".repeat(j/4));
					sb.append("BB".repeat((j%4)/2));
				}
				i+=j;
			}
			else {
				sb.append(".");
				i++;
		}
		}
		System.out.println(sb.toString());
	}
}