프로그래머스 – 이진 변환 반복하기 문제 Python

Programmers

Meet the best programming lessons in videos and practices. Programmers provide programing lessons, algorithm problems, programming competitions, and blockchain materials.

프로그래머스 문제풀이

문제

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

0과 1로 이루어진 어떤 문자열 x에 대한 이진 변환을 다음과 같이 정의합니다.

문제설명

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

입출력 예

입출력 예 설명

입출력 예 #1

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

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

입출력 예 #2

• “01110”이 “1”이 될 때까지 이진 변환을 가하는 과정은 다음과 같습니다.

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

입출력 예 #3

• “1111111”이 “1”이 될 때까지 이진 변환을 가하는 과정은 다음과 같습니다.

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

문제해결과정

1. S의 모든 0을 제거한다.
2. 0이 제거된 X의 길이를 이진변환하여 새로운 S로 만든다.
3. 1,2 과정을 반복한다.
4. 최종적으로 S가 1이 남았을때 이전변환 횟수와 제거한 0의 수를 출력한다.

위 같은 문제 및 제한사항을 해석하고 문제풀이를 시작하면 됩니다.

나의 풀이

저는 우선적으로 while True를 사용해서 무한반복하고 S의 길이가 이진변환해도 1이 될때 반복문을 종료하도록 코딩했어요.
그리고나서 0이 있다면 0을 제거하고 0이 없다면 이진변환을 하도록 해서 해결했습니다.

def solution(s):
    answer = [0,0]
    while True:
        if "0" in s:                           # 0이 있다면
            s = s.replace("0","",1)            # 0을 1번만 없앤다.
            answer[1] += 1
        else:                                  # 0이 없다면
            s = bin(len(s)).replace("0b","")   # 이진변환한다.
            answer[0] += 1
            if len(s) == 1:                    # 길이가 1이면 반복 종료
                break
    return answer

다른 사람의 풀이

파이써닉한 다른 풀이가 있어 가져와봤습니다.
문제의 본질을 잘 꽤뚫어본 코드가 아닌가 싶네요.
해당 풀이에서 이렇게 하나 배워갑니다. ㄷㄷ

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]