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

문제

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

2630번: 색종이 만들기

 

아이디어

1. 재귀 시점

색깔이 다르면 4등분으로 나눔

2. 종료 시점

4개로 나눈 것에 대한 탐색 함수 각각 실행 완료 후 종료

3. 결과 처리

조각을 다 돌고 같지 않은 색깔이 없는 경우 해당 조각의 색깔을 res 배열에 저장

res 배열 내 색깔별 갯수를 출력해 결과냄

 

전체 코드

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

import sys
input=sys.stdin.readline

N=int(input())
arr=[list(map(int, input().split())) for _ in range(N)]
#print(N,arr)
res=[]

def split(n,x,y):
    color=arr[x][y]
    #print(color)
    for i in range(x,x+n):
        for j in range(y,y+n):
            if arr[i][j]!=color:
                split(n//2,x,y)
                split(n//2,x,y+n//2)
                split(n//2,x+n//2,y)
                split(n//2,x+n//2,y+n//2)
                return
    if color==0:
       res.append(0)
    else:
       res.append(1)

split(N,0,0)
print(res.count(0))
print(res.count(1))