[프로그래머스] Swift - 이진 변환 반복하기(LV.2)

👆문제풀러 가기👆

문제 설명

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]

---

풀이 [ 메모리: 18,022kb, 최대시간: 16.5ms ]

func solution(_ s:String) -> [Int] {
    var s = s
    var binaryCnt = 0 // 이진변환 횟수
    var zeroCnt = 0 // 지워진 0의 갯수
    
    while s != "1" {
        let xLength = s.filter{$0 != "0"}.count // 1의 갯수
        let zero = s.count - xLength // 0의 갯수

        s = String(xLength, radix: 2) // 이진변환
        
        binaryCnt += 1
        zeroCnt += zero
    }
    
    return [binaryCnt, zeroCnt]
}

Swift에서는 기본 제공하는 진수변환 방법(String(value: BinaryInteger, radix: Int))이 존재한다. 

 

1. 0을 제거한 값의 글자수(xLength)를 계산
2. 제거된 0의 갯수 계산
3. 0을 제거한 값의 글자수(xLength)에 대해 이진변환
4. 이진변환 횟수 + 1
5. 제거된 0의 갯수 합산
6. 위의 과정을 s의 값이 1일 때까지 반복

 

Radix 없으면 어떻게 코드가 작성될지 궁금해서, 다시 작성해봤다.

다른 풀이 [ 메모리: 18,022kb, 최대 시간: 20.95ms ]

func solution(_ s:String) -> [Int] {
    var s = s
    var zeroCnt = 0
    var binaryCnt = 0
    
    while s != "1" {
        var c = s.filter{$0 != "0"}.count
        let zero = s.count - c
        var temp = "" // 반복마다 임시값 초기화
        
        // c가 0이 될 때까지 이진변환
        while c != 0 {
            temp = String(c % 2) + temp
            c /= 2
        }
        // 이진변환된 값을 s에 저장
        s = temp
        
        // 0의 갯수 합, 이진변환 횟수 +1
        zeroCnt += zero
        binaryCnt += 1
    }
    
    return [binaryCnt, zeroCnt]
}

최대 소요시간은 조금 더 걸리지만, 좀 더 기본에 충실한 느낌..