[Kotlin] 프로그래머스 - 이진 변환 반복하기
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)