[백준] / [Python] / [1343] 폴리오미노

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

---

문제

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

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

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

입력

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

출력

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

---

replace 사용 O

board = input().strip()

board = board.replace('XXXX', 'AAAA')
board = board.replace('XX', 'BB')

if 'X' in board: print('-1')
else: print(board)

replace 사용 X

board = input().strip()
board += '.'

i = 0
count = 0
answer = ''

while True:
    if board[i] == 'X': count += 1
    elif board[i] == '.':
        answer += '.'
        count = 0
    
    if i == len(board) - 1: break
    
    if (board[i] == 'X' and board[i+1] == '.'):
        div, mod = divmod(count, 4)
        if mod == 0:
            answer += 'AAAA' * div
        elif mod == 2:
            answer += 'AAAA' * div
            answer += 'BB'
        else:
            answer = '-1.'
            break
    
    i += 1

print(answer[:-1])