[백준] 1780번 종이의 개수 (파이썬)

1780번: 종이의 개수

 

코드

n = int(input())
li = []

for _ in range(n):
    li.append(list(map(int, input().split(" "))))

answer = []

def find(x, y, N):
    color = li[x][y]
    for i in range(x, x+N):
        for j in range(y, y+N):
            if li[i][j] != color:
                find(x,y, N//3)
                find(x, y+N//3, N//3)
                find(x, y+N//3*2, N//3)

                find(x+N//3,y, N//3)
                find(x+N//3,y+N//3, N//3)
                find(x+N//3,y+N//3*2, N//3)

                find(x+N//3*2,y, N//3)
                find(x+N//3*2,y+N//3, N//3)
                find(x+N//3*2,y+N//3*2, N//3)
                
                return
                
    if color == 0:
        answer.append(0)
    elif color == -1:
        answer.append(-1)
    else:
        answer.append(1)

find(0,0,n)
print(answer.count(-1))
print(answer.count(0))
print(answer.count(1))