문제

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의 개수를 세어야 한다 => 바로 collections의 Counter를 떠올리자.
  • 마찬가지로 남은 1의 개수를 계속 세주어야 하므로 Counter는 더 없이 좋은 선택
• 남은 1의 개수를 이진수로 바꾸기 위해 format(value, 'b')를 이용하면 된다.
• 위의 과정들을 s가 '1'이 될 때까지 반복하자.

코드

from collections import Counter
def solution(s):
    zero_count = 0
    iter = 0
        
    def convert(s):
        nonlocal zero_count
        counter = Counter(s)
        ones = counter['1']
        zero_count += counter['0']
        return format(ones, 'b')
        
    while s != '1':
        iter += 1
        s = convert(s)
    
    return [iter, zero_count]