[프로그래머스] 이진 변환 반복하기 (level2, python)

문제 설명

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 해야 합니다.

 

풀이 코드

def solution(s):
    answer = [0, 0]
    while s != '1':
        answer[0] += 1
        whole = len(s)
        s = ''.join(s.split('0'))
        one = len(s)
        answer[1] += (whole - one)
        s = bin(one)[2:]
        
    return answer

split 함수의 특성을 이용했다.

0을 모두 제거 하므로, 0을 기준으로 문자열을 나눠 다시 붙이면 1만 남은 문자열이 완성될 것이다.

문제의 특성상 1만 남은 문자열의 '길이'만 필요하므로, 굳이 split 말고 count를 활용하는 것이 더욱 효율적일 것이라고 생각된다.

 

제거한 0의 개수는, 제거하기 이전 길이 - 제거한 후 1로만 이뤄진 텍스트의 길이를 해서 구할 수 있다.

10진수를 2진수로 만드는 방법은 간단하다.

 

파이썬에서 제공하는 bin 함수를 사용하면 되는데,

bin 함수를 사용하면 앞에 '0b'라는 문자가 붙은 string type이 리턴된다.

따라서 슬라이싱을 통해 앞의 '0b'를 제거한 값을 s 에 갱신해준다.