[백준][Python] 색종이 만들기 - 재귀(분할정복)

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

2630번: 색종이 만들기

 

 

Python

import sys
input = sys.stdin.readline

n = int(input())
arr = [list(map(int, input().split())) for _ in range(n)]

def recur(r,c,size):
    global arr

    if size == 1:
        return str(arr[r][c])

    size //= 2
    ans = ''
    ans += recur(r,c,size)
    ans += recur(r+size,c,size)
    ans += recur(r,c+size,size)
    ans += recur(r+size,c+size,size)

    for a in ans:
        if a != ans[0]:
            return ans
    return str(arr[r][c])

ans = recur(0,0,n)
print(ans.count("0"))
print(ans.count("1"))