문제

🔗 월간 코드 챌린지 시즌1 | 이진 변환 반복하기

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

1. x의 모든 0을 제거합니다.
2. 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'이 최소 하나 이상 포함되어 있습니다.

내가 작성한 코드

import re
def solution(s):
    cnt, countz = 0, 0
    while(s != "1") :
        countz += s.count("0")
        s = bin(len(re.sub("[^1]","",s)))[2:]
        cnt += 1

    return [cnt, countz]

다음의 순서에 따라 풀이하였다.

1. 0의 개수를 셀 countz와 횟수를 셀 cnt 변수를 초기화 한다.

cnt, countz = 0,0

2. s가 1이 될 때 까지 3-4 과정을 반복한다.

while(s != "1") :

3. 문자열에 있는 "0"의 개수를 세어 countz에 담는다.

countz += s.count("0")

4. 문자열 s에서 0을 제거하고, 제거한 문자열의 길이를 측정해 2진수로 변환한다. 이때 변환된 s앞에 2진수를 뜻하는 0b가 붙으므로 인덱스 2번부터 잘라 s에 저장한다. 그리고 횟수를 1 증가시켜준다.

s = bin(len(re.sub("[^1]","",s)))[2:]
cnt += 1

5. 반환된 문자열이 1이라면 배열에 [cnt, countz]를 담아 리턴한다.

return [cnt, countz]

다른 사람의 풀이

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]

regex를 배웠으니 regex를 활용해서 최대한 간단한 문제를 풀어보았다! 더 복잡한 문제도 풀 수 있을때까지 👊