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

김가영·2021년 2월 10일
0

Algorithm

목록 보기
44/78

문제 바로가기

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 인것을 감안하면 시간 줄이기에 좋을듯

profile
개발블로그

0개의 댓글