1343 폴리오미노

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

1343번: 폴리오미노

 

내 풀이(맞음)

x = input()
count = 0
for i in range(len(x)):
    if(x[i]=='X'):
        count += 1
    
    if(count == 4 or (x[i]=='.' and count == 4)):
        x = x.replace('X', 'A', 4)
        count = 0
    elif((count == 2 and i == len(x)-1) or (x[i]=='.' and count == 2)):
        x = x.replace('X', 'B', 2)
        count = 0
    elif((x[i]=='.' and count % 2 == 1) or (count % 2 == 1 and i == len(x)-1)):
        count = -1
        break
print(x if (count != -1) else -1)

count에다가 X의 개수를 세준다. 그리고 첫번째 조건으로 count가 4개 이거나 현재 x[i]가 '.'이고 count가 4면 X 4개를 A로 바꾼다. 두번째 조건은 count가 2고 i가 마지막에서 2번째 이거나 현재 x[i]가 '.'이고 count가 2개면 X 2개를 B로 바꾼다. 여기서 마지막에서 2번째로 한건 마지막번째로 하면 범위 오류가 날 수 있기 때문이다. 마지막 세번째 조건으로 현재 x[i]가 '.'이고 count가 홀수거나 count가 홀수이고 i가 마지막에서 두번째이면 count에 -1을 넣고 break를 한다. 그리고 count가 -1이 아니면 X를 출력한다.