[Algorithm] Programmers : 이진 변환 반복하기 by Python

[문제 바로가기] https://programmers.co.kr/learn/courses/30/lessons/70129

📌문제 설명

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'이 최소 하나 이상 포함되어 있습니다.

입출력 예

s	result
"110010101001"	[3, 8]
"01110"	[3, 3]
"1111111"	[4, 1]

입출력 예 설명

입출력 예 #1

"110010101001"이 1이 될 때까지 이진 변환을 가하는 과정은 다음과 같습니다.

3번의 이진 변환을 하는 동안 8개의 0을 제거했으므로, [3,8]을 return 해야 합니다.

입출력 예 #2
"01110"이 1이 될 때까지 이진 변환을 가하는 과정은 다음과 같습니다.

3번의 이진 변환을 하는 동안 3개의 0을 제거했으므로, [3,3]을 return 해야 합니다.

---

💡문제 풀이

문제 설명의 길이만큼 간단한 문제!
이진수, 십진수 변환만 할 줄 알면 간다하게 해결할 수 있다.

s가 '1'이 될 때 까지 아래의 과정을 반복한다.

 

step 1) '0'의 개수를 센다.
step 2) '1'의 개수를 센다.
step 3) '1'의 개수(십진수)를 다시 이진수로 변환한다.
step 4) 이진 변환의 횟수에 +1을 한다.

 

• num_zero : '0'의 개수를 저장하는 변수
• num : '1'의 개수를 저장하는 변수
• cnt : 이진 변환의 횟수를 저장하는 변수

작성한 코드는 아래와 같다.

 

def solution(s):
    cnt = 0
    num_zero = 0
    while s != '1':
        num_zero += s.count("0")
        num = s.count('1')
        s = bin(num)[2:]
        cnt += 1
    return [cnt, num_zero]