[프로그래머스/Python] 이진 변환 반복하기

문제 정보

제목 : 이진 변환 반복하기

난이도 : Lv.2

사용 언어 : Python

문제 링크

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

 

문제 설명

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]

입출력 예 설명

입출력 예 #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 해야 합니다.

---

나의 풀이

문자열 s의 길이가 1이상이면 반복하는 while문을 생성했습니다. tmp변수에 count()함수를 이용하여 문자열 s의 0의 개수를 카운트하여 저장합니다. zero_count 변수에 더하기 할당 연산자(+=)를 이용해 tmp 값을 더합니다. 0 제거 후 길이를 구하기 위해 len(s) -tmp하고 sLen 변수에 저장합니다. while 반복문을 이용해 tmps에 sLen을 이진수로 변환한 값을 저장합니다. 이진 변환 결과를 구했다면 change_count에 1 증가시키고 tmps 문자열을 s에 저장합니다.

반복이 끝나면 answer배열에 추가하여 리턴합니다.

 

코드

def solution(s):
    answer = []
    change_count = 0
    zero_count = 0
    tmp = 0
    sLen = 0
    
    while ( len(s) > 1) :
        tmp = s.count('0')
        zero_count += tmp
        
        sLen = len(s) - tmp # 1의 개수 구함
        
        tmps =""
        while sLen > 0 :
            tmps = str(sLen%2) + tmps
            sLen//=2

        change_count +=1
        s = tmps
            
    answer.append(change_count)
    answer.append(zero_count)
    
    
    return answer