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

문제 바로가기

split(arr) : arr 을 네 개로 나누어 return 한다
check(arr) : arr 내의 0과 1의 개수를 세어 return 한다

Others

def count(arr, n):
    ones = sum(sum(arr, []))
    return n*n-ones, ones

arr에서 1을 세는 코드
sum(arr,[]) 를 통해 arr을 하나의 배열로 만들어 준 후, sum 을 통해 1의 개수를 가져온다

또, 나처럼 굳이 슬라이싱으로 배열을 나눠주지 않고 param 을 추가하여 index 처음과 끝만을 이용하기도 하더라. 슬라이싱이 시간복잡도 n 인것을 감안하면 시간 줄이기에 좋을듯

좋은 웹페이지 즐겨찾기