[algorithm] 백준 1343 - 폴리오미노 (파이썬)

📚 문제

입력

출력

예제 입력 1

XXXXXX

예제 출력 1

AAAABB

예제 입력 2

XX.XX

예제 출력 2

BB.BB

예제 입력 3

XXXX....XXX.....XX

예제 출력 3

-1

예제 입력 4

X

예제 출력 4

-1

예제 입력 5

XX.XXXXXXXXXX..XXXXXXXX...XXXXXX

예제 출력 5

BB.AAAAAAAABB..AAAAAAAA...AAAABB

🧑🏻‍💻 풀이 과정

• replace 함수를 이용해서 입력받은 문자 s의 XXXX를 AAAA로 변환하고, XX를 BB로 변환하자.
• if문을 이용해서 결과에 X가 포함되어 있으면 -1, 모두 변환에 성공했으면 결과를 출력하자.

def polyomino(s):
    replace_s = s.replace("XXXX", "AAAA").replace("XX", "BB")
    print(replace_s) if replace_s.count('X') == 0 else print(-1)


polyomino(input())