이진 변환 반복하기

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 solution(s):
    binary_list = list(s)
    turn, count_0 = 0, 0
    answer = []
    
    while s != "1":
        binary_list = list(filter(lambda x : int(x) == 1, s))
        count_0 += len(s) - len(binary_list)
        s = bin(len(binary_list))[2:]
        turn += 1
    
    answer.append(turn)
    answer.append(count_0)
    return answer

filter 함수를 사용해서 0을 제외한 리스트를 만들어 반복적으로 계산하는 방법을 사용

 

[좋은 풀이]

def solution(s):
    a, b = 0, 0
    while s != '1':
        a += 1
        num = s.count('1')
        b += len(s) - num
        s = bin(num)[2:]
    return [a, b]

turn의 경우 s != 1이기만 하면 무조건 1이 늘어나는 것이고, 결국 0이 아닌 1의 갯수가 다음 step의 s가 되는 셈이므로, 이 풀이가 정말 경제적이고 좋은 풀이라는 생각이 들었다.

나는 주어진 문제의 특징을 정확하게 파악하지 못한 것 같다.