이진 변환 반복하기(python)

in Algorithm on Algorithm, Programmers, 재귀, Python

참고자료

프로그래머스 - 이진 변환 반복하기

문제 설명

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

입출력 예제

sreturn
“110010101001”[3,8]
“01110”[3,3]
“1111111”[4,1]

입출력 예에 대한 설명

입출력 예 #1
문제의 예시와 같습니다.

  • “110010101001”이 “1”이 될 때까지 이진 변환을 가하는 과정은 다음과 같습니다.
회차이진 변환 이전제거할 0의 개수0 제거 후 길이이진 변환 결과
1“110010101001”66“110”
2“110”12“10”
3“10”11“1”
  • 3번의 이진 변환을 하는 동안 8개의 0을 제거했으므로, [3,8]을 return 해야 합니다.

입출력 예 #2

  • “01110”이 “1”이 될 때까지 이진 변환을 가하는 과정은 다음과 같습니다.
회차이진 변환 이전제거할 0의 개수0 제거 후 길이이진 변환 결과
1“01110”23“11”
2“11”02“10”
3“10”11“1”
  • 3번의 이진 변환을 하는 동안 3개의 0을 제거했으므로, [3,3]을 return 해야 합니다.

입출력 예 #3

  • “1111111”이 “1”이 될 때까지 이진 변환을 가하는 과정은 다음과 같습니다.
회차이진 변환 이전제거할 0의 개수0 제거 후 길이이진 변환 결과
1“1111111”07“111”
2“111”03“11”
3“11”02“10”
4“10”11“1”
  • 4번의 이진 변환을 하는 동안 1개의 0을 제거했으므로, [4,1]을 return 해야 합니다.

나의 생각

s가 “1”이 될까지 과정을 반복하기 때문에 재귀를 생각했다.
while문을 통해서 진행할 수도 있지만 재귀를 한번 구현해봤다.

솔루션

이진 변환 개수와 제거한 0의 개수를 계속해서 저장해야 되기 때문에 인수에 넣어준다.
종료 조건은 x == “1”일때 개수를 담은 list를 return 해준다.
x == “1”이 될때까지 계속 함수를 호출해준다.

코드

  1. 첫번째 코드
    • 재귀를 통해 반복
      def transform(x, trans_cnt, zero_cnt):
 if x == "1":
   return [trans_cnt, zero_cnt]
 trans_cnt += 1
 zero_cnt += x.count("0")
 x = bin(len(x) - x.count("0"))[2:]
 return transform(x, trans_cnt, zero_cnt)

      def solution(s):
 answer = transform(s, 0, 0)
 return answer
  2. 두번째 코드
    • while문을 통해 반복
      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]