[Programmers] 이진 변환 반복하기 (월간 코드 챌린지 시즌1 - Level2, Python3)

코딩테스트 연습 - 이진 변환 반복하기

 

문제 설명

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	result
"110010101001"	[3,8]
"01110"	[3,3]
"1111111"	[4,1]

 

 

힌트

간단한 구현 시뮬레이션 문제이다.

이 문제가 아직 어렵다면 문자열 처리 방법에 대해 조금 더 공부해보자.

 

Python에서 문자열은 변경할 수 없는 자료형이다.

1번 과정에서 모든 0을 제거하는 방법은 정말 여러 가지가 있지만 잘 모르겠다면 문자열 내장 함수인 replace 함수에 대해 알아보자.

2번 과정에서 2진법으로 표현하는 방법으로 bin 함수를 사용하는 방법이 있다.

 

 

풀이

문제에서 시키는대로 하면 된다.

힌트에서 얘기했던 방식대로 replace 함수와 bin 함수를 사용하여 간단하게 문제를 해결할 수 있다.

 

def solution(s):
    count = 0
    zero_remove = 0
    
    # s가 '1'이 될 때까지 반복
    while s != '1':
        # 제거된 0의 개수 누적
        zero_remove += s.count('0')
        # 1. 문자열 s에서 0을 제거
        s = s.replace('0', '')
        # 2. s의 길이를 c라고 하면,
        # s를 "c를 2진법으로 표현한 문자열"로 바꿉
        c = len(s)
        s = bin(c)[2:]

        count += 1
    
    answer = [count, zero_remove]
        
    return answer