[프로그래머스/Python] Lv. 2 이진 변환 반복하기

안녕하세요.

이번 문제는 "이진 변환 반복하기"입니다.

 

문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/70129

프로그래머스

 

문제 요약

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

1. x의 모든 0을 제거합니다.
2. x의 길이를 c라고 하면, x를 "c를 2진법으로 표현한 문자열"로 바꿉니다.

ex) x = "0111010"이라면, x에 이진 변환을 가하면 x = "0111010" -> "1111" -> "100" 이 됩니다.

 

0과 1로 이루어진 문자열 s가 매개변수로 주어집니다.

결론 : s가 "1"이 될 때까지 계속해서 s에 이진 변환을 가했을 때, [이진 변환의 횟수]와 [변환 과정에서 제거된 모든 0의 개수]를 각각 배열에 담아 return 하도록 solution 함수를 작성하는 것입니다.

 

문제 풀이

🔍 제가 생각한 풀이 방법은,

1. 먼저 문자열 s에서 제거할 0의 개수를 센다.

2. 문자열 s를 for문을 돌면서 0이 아닐 때 즉, 값이 1인 것들만 리스트에 하나씩 담는다.

3. 그 리스트의 길이를 2진법 문자열로 바꾼다.

4. 위와 같은 이진 변환 과정의 횟수를 담는 변수에 +1을 한다.

• 1~4 과정을 문자열 s가 1이 될 때까지 반복한다.

5. 마지막으로, 리스트에 각 구한 값을 리스트에 담아 return 한다.

 

✋ 여기서 잠깐!

리스트의 길이를 2진법 문자열로 바꿀 때 문자열에 0b가 붙어서 출력되는데 이걸 없애기 위해 bin(len(s))[2:] 이렇게 문자열을 인덱스로 자르는 방법을 써야 합니다.

 

전체 코드

def solution(s):
    cnt1, cnt2 = 0, 0
    
    while s != "1":
        cnt2 += s.count("0")
        s = [i for i in s if i not in "0"]
        s = bin(len(s))[2:]
        cnt1 += 1
    
    return [cnt1, cnt2]