[Swift] 알고리즘 공부 - 이진 변환 반복하기

문제 설명

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	result
"110010101001"	[3, 8]
"01110"	[3, 3]
"1111111"	[4, 1]

 

 

설명

1. 먼저 s가 1이 될 때까지 0을 없애는 작업과, 2진수로 변환하는 작업을 반복해야 하기 때문에 반복문을 사용한다.

2. changeS라는 변수에 s값을 저장하여 for문을 돌린다. 이 때 i값이 1이면 convertS에 저장하고 아니면 eraseCount(0을 몇번지웠는지 저장할 변수) count를 올려준다.

3. for문을 돌려서 나온 convertS의 자릿수만큼 2진수로 변환

4. 2진수로 변환된 값을 changeS 변수에 저장하고 총 count수를 +1해준다,

5. changeS가 1이 될때까지 반복

import Foundation

func solution(_ s:String) -> [Int] {
    var changeS = s
    var convertS = ""
    var eraseCount = 0
    var count = 0
    
    while changeS != "1" {
        for i in changeS {
            if i == "1" { convertS += String(i) }
            else { eraseCount += 1 }
        }
        convertS = String(convertS.count, radix: 2)
        changeS = convertS
        convertS = ""
        count += 1
    }
    return [count, eraseCount]
}