[BOJ 1343, Python 3] 폴리오미노

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

1343번: 폴리오미노

 

 

풀이

사전순으로 가장 앞서는 답은 A가 먼저 나와야한다는 것이다.

그래서 연속된 X의 수가 4개씩 나올 때마다 바로바로 저 4칸을 A로 바꾸어 주고, '.'이 나왔거나 마지막 문자까지 다 확인했는데 연속된 X의 수가 2개이면 2칸을 BB로 바꾸면 된다.

 

만약 연속된 X의 수가 홀수개인데 '.'이 나온다면 바로 -1을 출력하고 종료해주면 된다.

 

코드

arr = [*input()]
ans = ['.'] * len(arr)
 
val = 0
for i in range(len(arr)):
    if arr[i] == 'X': val += 1
    if arr[i] == '.' and val % 2: print(-1); exit()
    if arr[i] == '.' and val == 2: ans[i-2:i] = ['B']*2; val = 0
    if val == 4: ans[i-3:i+1] = ['A']*4; val = 0
if val == 2: ans[-2:] = ['B']*2
print([''.join(ans), -1][val % 2])