[프로그래머스] LV.2 이진 변환 반복하기(python)

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

프로그래머스

문제 설명

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 함수를 완성해주세요.

 

문제 해결

def solution(s):
    answer = []
    total = 0
    zero = 0
    while(s != '1'):
        total += 1
        num = s.count('1')
        zero += (len(s) - num) #남은 건 '0'의 개수
        s = bin(num)[2:]
    return [total, zero]

 

문제 풀이 설명

먼저 문자열 s가 '1'이 될때까지 반복하고, 제거한 '0'의 개수를 저장하고, 문자열 s를 '0'을 제거한 s의 길이를 다시 이진수로 표현해 s를 update 해줘야 하는게 문제의 목적이다.

total 변수로 몇 번 update 해줬는지, zero 변수로 제거한 '0'의 개수를 저장하고 따로 s에서 '0'을 제거할 필요 없이 '1'의 개수를 count() 함수로 세어줘 num 변수에 저장했다.

bin() 함수를 사용해 2진수로 변환 후 '0b'가 붙는걸 슬라이싱으로 지워줬다.