[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'이 최소 하나 이상 포함되어 있습니다.

def func_bin(x):
    li = []
    res = ''
    
    while(x):
        li.append(x%2)
        x = x // 2
    
    for i in range(len(li)-1, -1, -1):
        res += str(li[i])
    
    return res

def solution(s):
    answer = []
    cnt_z_tot = 0
    cnt_turn = 0
    
    while(s != "1"):
        cnt_z = 0
        for i in s:
            if i == '0':
                cnt_z += 1
        cnt_z_tot += cnt_z
        bin = len(s)-cnt_z # 0 제거 후 길이
        s = func_bin(bin)
        cnt_turn += 1 
    
    return [cnt_turn, cnt_z_tot]

• 문제는 간단하다. 문자의 모든 0을 제거한 뒤 제거 후의 길이를 2진수로 변환하고 이 과정만 계속해서 반복하면 된다.
• 문제를 풀기 위해서 func_bin(int num) 이라는 숫자(문자 길이)를 받아와서 이진변환을 하고 결과를 다시 문자열로 돌려주는 함수를 만들었다.
• 그리고 이진 변환 과정에서 최종 문자가 1이 되기 전까지 0을 제거하고 이진변환하는 과정을 반복하도록 하였다.
• 조금 더 간단하게 풀 수 있는 방법이 있지 않을까 싶은데 더 고민해봐야 겠다.