[프로그래머스_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'이 최소 하나 이상 포함되어 있습니다.

 

입출력 예

 

 

풀이

class Solution {
    fun solution(s: String): IntArray {
        var answer: IntArray = IntArray(2)
        var str = s

        while (str.length != 1) {
            //횟수 추가
            answer[0]++
            //0 개수 더하기
            answer[1] += str.count { it == '0' }
            //0 제거
            str = str.replace("0", "")
            //2진법으로 바꾸기
            str = Integer.toBinaryString(str.length)
        }
        return answer
    }
}

 

자꾸 시간초과로 떠서 뭐지했는데

while문에서 str를 초기화하지 않고 써서 그랬었다.

 

 

다른 사람의 풀이

class Solution {
    fun solution(s: String): IntArray {
        var copiedS = s
        var removedZero = 0
        var count = 0

        while (copiedS != "1") {
            removedZero += copiedS.replace("1", "").count()
            copiedS = Integer.toBinaryString(copiedS.replace("0", "").count())
            count++
        }
        return intArrayOf(count, removedZero)
    }
}

 

 

 

풀이에서 2진법으로 바꾸는 코드 개선

str = Integer.toBinaryString(str.replace("0", "").length)