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

문제

---

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

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

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

입력

---

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

출력

---

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

예제 입력 1

---

XXXXXX

예제 출력 1

---

AAAABB

예제 입력 2

---

XX.XX

예제 출력 2

---

BB.BB

나머지 예제는 생략한다.

풀이

---

Python

import sys
input = sys.stdin.readline

s = input().rstrip()

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

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

정리

---

• 처음에 문자열을 돌면서 X의 개수를 카운트 해주면서 답을 찾아가려 했으나 코드가 복잡해져 replace함수를 이용해 문제를 해결했다.