[Python] 쿼드압축 후 개수 세기

https://school.programmers.co.kr/learn/courses/30/lessons/68936

---

압축을 하는 quad 함수를 외부에 선언할 경우 answer를 global로 선언해서 quad 함수에서도 쓸 수 있게 해야하고 arr를 파라미터로 줘야한다.

 

def solution(arr):
    global answer
    answer = [0,0]  # 0의 개수, 1의 개수
    quad(arr, answer, len(arr))
    return answerdef quad(arr, s, n):
    x, y, target = s[0],s[1], arr[s[0]][s[1]]
    for i in range(n):
        for j in range(n):
            if target != arr[x+i][y+j]:
                n //= 2
                quad(arr, [x,y],n)
                quad(arr, [x,y+n], n)
                quad(arr, [x+n,y], n)
                quad(arr, [x+n,y+n], n)
                return
    answer[target] += 1

 

개인적으로 global을 잘 안 쓰다 보니 quad 함수를 내부에 선언할 경우가 더 편했다

근데 파이썬 알고리즘 인터뷰 책을 읽고 나니 내부함수명에 _를 안 붙이면 규칙에 어긋나는 거 같아서 신경쓰인다

def solution(arr):
    def quad(x,y,n):
        target = arr[x][y]
        for i in range(n):
            for j in range(n):
                if target != arr[x+i][y+j]:
                    n //= 2
                    quad(x,y,n)
                    quad(x+n,y,n)
                    quad(x,y+n,n)
                    quad(x+n,y+n,n)
                    return
        answer[target] += 1
    answer = [0,0]
    n = len(arr)
    quad(0,0,n)
    return answer