split(arr)
: arr 을 네 개로 나누어 return 한다
check(arr)
: arr 내의 0과 1의 개수를 세어 return 한다
def count(arr, n):
ones = sum(sum(arr, []))
return n*n-ones, ones
arr에서 1을 세는 코드
sum(arr,[])
를 통해 arr
을 하나의 배열로 만들어 준 후, sum 을 통해 1의 개수를 가져온다
또, 나처럼 굳이 슬라이싱으로 배열을 나눠주지 않고 param 을 추가하여 index 처음과 끝만을 이용하기도 하더라. 슬라이싱이 시간복잡도 n 인것을 감안하면 시간 줄이기에 좋을듯