인프런 커뮤니티 질문&답변

기본 로직은 큰돌님과 같이
1. 처음에 0 이 있다면 앞에서 모두 뺴기
2. 비교는 size -> String 각각 비교로 진행하였습니다.


위의 코드는 String 을 Array 로 받은 후, 직접 0을 걸렀고

밑의 코드는 내부에서 0을 걸렀습니답.

둘 다 예제나 스스로 만들어본 반례 모두 통과하는데, 위의 코드는 통과하고 밑의 코드는 실패해서
언어의 특성때문인지, 제가 놓치는 알고리즘적인 요소가 있는지 궁금해서 여쭤봅니다.

* 그리고 혹시, 이렇게 다른 언어를 여쭤봐도 괜찮은가요 ..?
특히 Swift 로 다시 풀어볼 때 많이 틀리는 것 같아서 질문 올려봅니답 ㅠㅠ..

통과코드 :

var testCase = Int(readLine()!)!
var result = [String]()
var isZero = false

while testCase > 0 {
    let input = Array(readLine()!)
    var integer = ""
    
    for i in 0 ..< input.count {
        if integer == "" && i != input.count - 1 && input[i] == "0" && (input[i + 1] >= "0" && input[i + 1] <= "9")  {
            continue
        }
        
        if input[i] >= "0" && input[i] <= "9" {
            integer += String(input[i])
            
            if i == input.count - 1 {
                result.append(integer)
            }
            else if input[i + 1] > "9" || input[i + 1] < "0" {
                result.append(integer)
                integer = ""
            }
        }
    }
    
    testCase -= 1
}

result.sort{ (first, second) -> Bool in
    if first.count == second.count {
        return first < second
    } else {
        return first.count < second.count
    }
}

result.forEach{
    print($0)
}

실패코드:

let N = Int(readLine()!)!
var result = [String]()

for _ in 0 ..< N{
    let input = readLine()!
    var tempValue = ""
    
    for i in input.indices {
        
        switch input[i].asciiValue! {
        case Character("0").asciiValue! ... Character("9").asciiValue!:
            if i == input.index(before: input.endIndex) {
                while !tempValue.isEmpty && tempValue.first == "0" {
                    tempValue.removeFirst()
                }
                result.append(tempValue + String(input[i]))
                continue
            }
            tempValue += String(input[i])

        default:
            if tempValue == "" {
                continue
            }
            
            while !tempValue.isEmpty && tempValue.first == "0" {
                tempValue.removeFirst()
            }
            result.append(tempValue)
            tempValue = ""
        }
    }
}

result.sorted {
    $0.count == $1.count ? $0 < $1 : $0.count < $1.count
}.forEach {
    print($0)
}