백준 1343번: 폴리오미노 (파이썬)

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

1343번: 폴리오미노

 

문제

 

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

 

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

 

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

 

입력

 

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

 

출력

 

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

 

풀이

 

보드의 앞에서부터 확인하며 폴리오미노를 채우도록 한다.

사전 순으로 배치되어야 하기 때문에 AAAA, BB 순서로 채우도록 한다.

 

소스 코드

 

board = input()

ans = []
i = 0
while i < len(board):
    if board[i:i+4] == 'XXXX':
        ans.append('AAAA')
        i += 4
    elif board[i:i+2] == 'XX':
        ans.append('BB')
        i += 2
    elif board[i] == '.':
        ans.append('.')
        i += 1
    else:
        break

ans = ''.join(ans)
if len(board) == len(ans):
    print(ans)
else:
    print(-1)