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

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

1780번: 종이의 개수

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


def counting_papers(r, c, n):

    check = paper[r][c]
    flag = False
    for i in range(r, r+n):
        for j in range(c, c+n):
            if paper[i][j] != check:
                flag = True
                break
        if flag:
            break
    result = [0, 0, 0]

    if flag:
        for k in range(3):
            for l in range(3):
                tmp = counting_papers(r+n//3*k, c+n//3*l, n//3)
                result = [result[i] + tmp[i] for i in range(3)]
        return result

    if check == -1:
        return [1,0,0]
    elif check == 0:
        return [0,1,0]
    else:
        return [0,0,1]


res = counting_papers(0,0,N)
for answer in res:
    print(answer)