PS/프로그래머스

이진 변환 반복하기 (lv2)

ForteQook 2022. 8. 13. 17:23

문제 설명

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

 문자열 s에서 0을 모두 제거한 문자열의 길이는 s에 포함된 1의 개수와 같다. 또한 s의 1의 개수가 2의 지수 배 만큼이 된다면, 다음 "이진 변환"에서 s가 1이 될 것이다. 이를 구현하면 다음과 같다.

from math import log2

def solution(s):
    answer = [0,0]
    while True:
        l1,l2 = len(s),s.count('1')
        
        if log2(l2) == int(log2(l2)):
            answer[0] += 2 if l2 > 1 else 1
            answer[1] += int(log2(l2)) + (l1-l2)
            break
        else:
            s = format(l2,'b')
            answer[0] += 1
            answer[1] += l1-l2
    return answer

'PS > 프로그래머스' 카테고리의 다른 글

줄 서는 방법 (lv2)  (0) 2022.08.14
[3차] 방금그곡 (lv2)  (0) 2022.08.13
3 x n 타일링 (lv2)  (0) 2022.08.13
예상 대진표 (lv2)  (0) 2022.08.12
2개 이하로 다른 비트 (lv2)  (0) 2022.08.12