[백준] 2630번 색종이 만들기 (파이썬)

2630번: 색종이 만들기

 

 

코드

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//2)
                find(x, y+N//2, N//2)
                find(x+N//2, y, N//2)
                find(x+N//2, y+N//2, N//2)
                return
                
    if color == 0:
        answer.append(0)
    else:
        answer.append(1)

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