백준 2630 - 색종이 만들기 (파이썬)

Answer

import sys

N = int(sys.stdin.readline())
paper = [list(map(int, sys.stdin.readline().split())) for _ in range(N)] 

result = []

def solution(x, y, N):
  color = paper[y][x]

  for i in range(x, x + N):
    for j in range(y, y + N):
      if color != paper[j][i]:
        solution(x, y, N // 2)
        solution(x, y + N // 2, N // 2)
        solution(x + N // 2, y, N // 2)
        solution(x + N // 2, y + N // 2, N // 2)
        
        return

  if color == 0 :
    result.append(0)
  else :
    result.append(1)

solution(0,0,N)

print(result.count(0))
print(result.count(1))