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

문제

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"이 될 때까지 이진 변환을 가하는 과정은 다음과 같습니다.

회차이진 변환 이전제거할 0의 개수0 제거 후 길이이진 변환 결과

1	"110010101001"	6	6	"110"
2	"110"	1	2	"10"
3	"10"	1	1	"1"

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

입출력 예 #2

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

회차이진 변환 이전제거할 0의 개수0 제거 후 길이이진 변환 결과

1	"01110"	2	3	"11"
2	"11"	0	2	"10"
3	"10"	1	1	"1"

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

입출력 예 #3

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

회차이진 변환 이전제거할 0의 개수0 제거 후 길이이진 변환 결과

1	"1111111"	0	7	"111"
2	"111"	0	3	"11"
3	"11"	0	2	"10"
4	"10"	1	1	"1"

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

 

✔풀이

def solution(s):
    answer = [0,0]
    
    for i in range(len(s)):
    
        if s!="1":
            answer[1]+=s.count("0")
            s=len(s.replace("0",""))
            s=bin(s)[2::]
            answer[0]+=1
            
        else:
            return answer

📝설명

출력을 리스트 형태로 해야 해서 애초에 answer을 [0,0]으로 초기화한 후 코드를 작성했다.

while문보다는 for문이 시간제한에 있어서 유리하다고 생각해서 반복횟수는 대충 입력받은 s의 길이로 했다.

출력되는 값에 대한 예시를 잘 보면 answer[0]은 for문의 반복 횟수, answer[1]은 제거한 0의 개수라는 걸 알 수 있다.

이에 따라 s가 1이 아닐 경우, answer[1]에는 s에서 "0"을 센 개수를 더해줬다.

그 다음에 0을 제거한 후 길이를 구할 수 있도록 replace와 len을 이용하여 s를 재정의했다.

다음 반복을 하기 위해선, s가 이진수로 표현되어야 하기 때문에 bin과 슬라이싱을 활용하여 s를 다시 정의했다.

[2::]을 한 이유는 bin을 통해 나오는 이진수는 0b를 앞에 포함하기 때문에 이 부분을 제외하기 위해서이다.

그리고 answer[0]에 1을 더하면 된다.

이렇게 반복하다가 s가 1이 된다면 else로 빠지면서 결과가 출력된다.

 

👑모범 답안

🔎 0을 카운트하지 않고 1을 카운트해서 진행했다.

 

💻 문제 출처

https://school.programmers.co.kr/learn/courses/30/lessons/70129