간만의 알고리즘 공부 - 4 (이진 변환 반복하기)

문제 설명
0과 1로 이루어진 어떤 문자열 x에 대한 이진 변환을 다음과 같이 정의합니다.
- x의 모든 0을 제거합니다.
- 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~2가지 테스트케이스만 통과하고 다 실패했다. 시간이 꽤나 흘러가기에 일단 코드 첨삭을 진행

 

도움을 받은 코드

fun solution(s: String): IntArray {
    var countOfTransformations = 0
    var totalZeroesRemoved = 0
    var currentString = s

    while (currentString != "1") {
        val zeroesCount = currentString.count { it == '0' }
        totalZeroesRemoved += zeroesCount
        val lengthAfterRemoval = currentString.length - zeroesCount
        currentString = lengthAfterRemoval.toString(2)
        countOfTransformations++
    }

    return intArrayOf(countOfTransformations, totalZeroesRemoved)
}

 

생각보다 간단하게 풀림에 놀라고 제대로 못풀었음에 조금 실망했다.

 

 

count() 내장함수

String 의 특정 문자를 카운트 할 수 있는 내장 함수 .count { it == '문자' }

public inline fun CharSequence.count(predicate: (Char) -> Boolean): Int {
    var count = 0
    for (element in this) if (predicate(element)) ++count
    return count
}

 

 

toString(radix: Int) 숫자를 진법으로 변환

val number = 42
val binaryString = number.toString(2) // 이진법 변환
println(binaryString) // 출력: 101010


val number = 255

val binaryString = number.toString(2) // 이진법 변환
println(binaryString) // 출력: 11111111

val octalString = number.toString(8) // 팔진법 변환
println(octalString) // 출력: 377

val hexString = number.toString(16) // 십육진법 변환
println(hexString) // 출력: ff