[Programmers 프로그래머스] 70129 이진 변환 반복하기 (PYTHON)

https://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문을 만들어 문제를 해결했습니다.

먼저 횟수를 세어줄 cnt와 지운 0의 수를 저장할 zero_cnt를 만들어 줍니다.

먼저 s에서 1과 0을 분리를 해주고, 1은 temp_s에 모아주고, 0은 zero_cnt에 넣어줍니다. 그렇게 만들어진 temp_s의 길이로 다시 이진수로 변환을 해서 s로 변경합니다.

이러한 과정을 반복해서 결과를 만들어 줍니다.

def solution(s):
    cnt = 0
    zero_cnt = 0
    while s != "1":	# s가 1이 될때 까지
        temp_s = ""
        for i in s:
            if i == "1":
                temp_s += i
            else:
                zero_cnt += 1
        s = format(len(temp_s), 'b') # 2진법으로 변환
        cnt += 1
    answer = [cnt, zero_cnt]
    return answer