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

문제

링크

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

• 입력: "110010101001"
• 출력: [3,8]

입출력 예 #2

• 입력: "01110"
• 출력: [3,3]

입출력 예 #3

• 입력: "1111111"
• 출력: [4,1]

코드 1

class Solution {
    fun solution(s: String): IntArray {
        var target = s
        var removeCount = 0
        var convertCount = 0
        while(target != "1") {
            convertCount++
            removeCount += target.count { it == '0' }
            target = Integer.toBinaryString(target.replace("0", "").length)
        }
        return intArrayOf(convertCount, removeCount)
    }
}

테스트 1 〉	통과 (11.32ms, 60.6MB)
테스트 2 〉	통과 (32.29ms, 62.8MB)
테스트 3 〉	통과 (8.24ms, 61.2MB)
테스트 4 〉	통과 (8.14ms, 60.2MB)
테스트 5 〉	통과 (8.04ms, 61.4MB)
테스트 6 〉	통과 (10.04ms, 60.9MB)
테스트 7 〉	통과 (8.84ms, 63.2MB)
테스트 8 〉	통과 (10.01ms, 63MB)
테스트 9 〉	통과 (16.71ms, 62MB)
테스트 10 〉	통과 (25.73ms, 61.5MB)
테스트 11 〉	통과 (27.83ms, 61.9MB)

코드 2. 속도 개선 (replace 함수 미사용)

class Solution {
    fun solution(s: String): IntArray {
        var target = s
        var removeCount = 0
        var convertCount = 0
        while(target != "1") {
            convertCount++
            target.count { it == '0' }.apply {
                removeCount += this
                target = Integer.toBinaryString(target.length - this)
            }
        }
        return intArrayOf(convertCount, removeCount)
    }
}

테스트 1 〉	통과 (0.04ms, 62MB)
테스트 2 〉	통과 (7.91ms, 60.8MB)
테스트 3 〉	통과 (0.05ms, 61.8MB)
테스트 4 〉	통과 (0.05ms, 62.7MB)
테스트 5 〉	통과 (0.03ms, 61.8MB)
테스트 6 〉	통과 (0.08ms, 61.1MB)
테스트 7 〉	통과 (0.13ms, 62.2MB)
테스트 8 〉	통과 (0.13ms, 59.8MB)
테스트 9 〉	통과 (4.89ms, 59.4MB)
테스트 10 〉	통과 (9.02ms, 60.1MB)
테스트 11 〉	통과 (5.83ms, 60.5MB)