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

문제

민식이는 다음과 같은 폴리오미노 2개를 무한개만큼 가지고 있다. AAAA와 BB
이제 '.'와 'X'로 이루어진 보드판이 주어졌을 때, 민식이는 겹침없이 'X'를 모두 폴리오미노로 덮으려고 한다. 

이때, '.'는 폴리오미노로 덮으면 안 된다.
폴리오미노로 모두 덮은 보드판을 출력하는 프로그램을 작성하시오.

 

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

입력

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

출력

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

예제

# input
XXXXXX
# output
AAAABB

# input
XX.XX
# output
BB.BB

# input
XXXX....XXX.....XX
# output
-1

문제 풀이

# 입력 받기
board = input().split('.')

# 정답 변수 초기화
answer = []

# 플래그 변수 초기화
is_valid = True

for i in board:
    # 부분 문자열의 길이가 홀수인 경우 유효하지 않음
    if len(i) % 2 != 0:
        is_valid = False
        break
    else:
        # 변환 결과를 임시로 저장할 변수 초기화
        tmp = ''
        
        # 부분 문자열의 길이가 4의 배수인 부분을 'AAAA'로 변환
        tmp += 'AAAA' * (len(i) // 4)
        
        # 남은 길이가 2인 경우 'BB'로 변환
        if len(i) % 4 == 2:
            tmp += 'BB'
            
        answer.append(tmp)
        
# 모든 부분 문자열이 유효한 경우
if is_valid:
    print('.'.join(answer))
else:
    print(-1)

# 입력 받기
board = input()

# 'XXXX'를 'AAAA'로, 'XX'를 'BB'로 교체
board = board.replace('XXXX', 'AAAA')
board = board.replace('XX', 'BB')

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

---

파이썬을 독학하시는 분들에게 도움이 되길 바라며,

혹 더 좋은 방법이 있거나 오류가 있다면 편하게 말씀 부탁드립니다.