[프로그래머스] 이진변환 반복하기 by Python

문제

0과 1로 이루어진 어떤 문자열 x에 대한 이진 변환을 다음과 같이 정의합니다.

x의 모든 0을 제거합니다.
x의 길이를 c라고 하면, x를 "c를 2진법으로 표현한 문자열"로 바꿉니다.
예를 들어, x = "0111010"이라면, x에 이진 변환을 가하면 x = "0111010" -> "1111" -> "100" 이 됩니다.

0과 1로 이루어진 문자열 s가 매개변수로 주어집니다. s가 "1"이 될 때까지 계속해서 s에 이진 변환을 가했을 때, 이진 변환의 횟수와 변환 과정에서 제거된 모든 0의 개수를 각각 배열에 담아 return 하도록 solution 함수를 완성해주세요.

제한사항

s의 길이는 1 이상 150,000 이하입니다.
s에는 '1'이 최소 하나 이상 포함되어 있습니다.

입출력

풀이

문자열을 받았을 때 0을 제거한 후,
제거한 리스트의 길이를 이진수로 표현,
그 이진수에서 다시 0을 제거,
제거한 리스트의 길이를 이진수로 표현
이렇게 반복하다가 "1"이 나올 때 멈춘다.

def solution(s):
	cnt = 0 # 이진변환 반복 count
    zeros = 0 # 제거되는 0의 갯수
    
    while True:
    	if s == '1':
        	break
        zeros += s.count('0') # 0의 갯수 count
        s = s.replace('0','') # 0 제거
        s = bin(len(s))[2:] # 0이 제거된 리스트의 길이를 이진수로 변환
        
        cnt = cnt + 1 # 이진변환 count
    answer = [cnt, zeros]
    
    return answer