[Lv. 2] 이진 변환 반복하기

문제 출처 : https://school.programmers.co.kr/learn/courses/30/lessons/70129

프로그래머스

난이도 : Level.2

언어 : Kotlin

 

문제 설명 :

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	return
"110010101001"	[3, 8]
"01110"	[3, 3]
"1111111"	[4, 1]

풀이 : 

class Solution {
    fun solution(s: String): IntArray {
        val answer: IntArray = intArrayOf(0, 0)

        var str = s
        while (true) {
            if (str == "1") break
            val cnt = str.count { it == '0' }

            str = str.count { it == '1' }.toString(2)

            answer[0] += 1
            answer[1] += cnt
        }

        return answer
    }
}