[Swift - 프로그래머스] [3차] 파일명 정렬

🕵🏻  문제 해석

---

파일 저장소 서버 관리를 맡게 된 무지가 기존에 이름 순으로 정렬된 파일들을 다시 보기 좋게 정렬하려고 합니다. 파일명은 크게 3 가지로 구성되는데 HEAD, NUMBER, TAIL입니다. 각각의 특징은 다음과 같습니다.

파일명에서 먼저 HEAD가 등장하며 그 이후 0에서 최대 99999까지의 NUMBER가 등장합니다. 이후 TAIL이 등장하지만 정렬기준에는 존재하지 않기 때문에 생각하지 않겠습니다. 정렬 우선순위는 다음과 같습니다.

• HEAD 기준으로 사전순으로 정렬한다. 이때 대소문자는 구별하지 않습니다.
• HEAD가 동일할 경우, NUMBER를 기준 오름차순 정렬합니다. 
• HEAD와 NUMBER가 모두 동일하다면, 입력순서대로 정렬합니다.

따라서 파일명이 주어지면 HEAD와 NUMBER 그리고 입력 순서 이 세 가지를 뽑아내서 배열에 저장한 후, 정렬 규칙에 맞게 정렬하여 리턴하는 방식으로 접근해보았습니다. 먼저 파일명을 String 타입 메서드인 .components를 이용하여 숫자를 제외한 문자열들로 나누어 주고 그 첫 번째 원소를 HEAD로 추출했습니다.

let numbers: CharacterSet = ["0","1","2","3","4","5","6","7","8","9"]

for idx in 0..<files.count {
        let fileComponents =  files[idx].components(separatedBy: numbers)
        var temp = [String]()
        
        for component in fileComponents {
            if component != "" {
                temp.append(component)
            }
        }
        
        let head = temp[0]
}

그 후, 본래 파일명에서 HEAD를 제거하면 다시 문자가 등장하기 전까지 숫자들을 NUMBER라고 판단할 수 있습니다. String 타입메서드중 .replacingOccurrences() 를 사용하여 HEAD를 제거한 후, 숫자를 구해보았습니다. 

그 후, HEAD의 대소문자 구별을 없애기위해, 대문자로 변환하고, 앞서 구한 "00100"처럼 앞에 0을 포함한 숫자들은 Int() 생성자를 통해 0을 제거한 Int 값으로 NUMBER를 구합니다. 이를 (String, Int, Int)  튜플에 저장하고 배열에 담아 정렬하여 정답을 구합니다.

📦  문제 풀이 & 코드

---

func solution(_ files:[String]) -> [String] {
    let numbers: CharacterSet = ["0","1","2","3","4","5","6","7","8","9"]
    let numbersString: [Character] = ["0","1","2","3","4","5","6","7","8","9"]
    var headNumberInputOrder = [(head: String, number: Int, idx: Int)]()
    var answer = [String]()
    
    for idx in 0..<files.count {
    
        // file[idx] = "foo010bar020.zip"
        let fileComponents =  files[idx].components(separatedBy: numbers)
        var temp = [String]()
        
        // fileComponents = ["foo","" ,"" ,"" ,"bar" ,"" ,"" ,"" ,".zip"]
        for component in fileComponents {
            if component != "" {
                temp.append(component)
            }
        }
        // temp = ["foo", "bar", ".zip"]
        
        let head = temp[0]
        
        let removeHead = files[idx].replacingOccurrences(of: head, with: "")
        // removeHead = "010bar020.zip"
        var number = ""
        
        for nums in removeHead {
            if numbersString.contains(nums) {
                number += String(nums)
            } else {
                break
            }
        }
        
        // ("FOO", 10, 0) 로 저장 (대문자화한 HEAD, NUMBER, 입력순서)
        
        headNumberInputOrder.append((head.uppercased(), Int(number)!, idx))
    }
    
    // 기준에 맞게 정렬
    
    headNumberInputOrder.sort {
        if $0.head == $1.head {
            if $0.number == $1.number {
                return $0.idx < $1.idx
            }
            return $0.number < $1.number
        }
        return $0.head < $1.head
    }
    
    // 배열을 탐색하며 answer에 정렬 순서대로 저장
    
    for tuple in headNumberInputOrder {
        answer.append(files[tuple.2])
    }
    
    return answer
}

 

🤷🏻‍♂️  더 생각해보기

---

제 풀이는 좀 지저분 하게 HEAD와 NUMBER를 구한 것 같습니다. 다른 분들의 풀이를 참고하니 훨씬 더 쉽고 간편한 풀이 법들이 있더군요. 아직 갈 길이 먼 것 같습니다.. 크게 HEAD와 NUMBER를 구할 때 정규식을 이용하여 구하는 방법과 drop과 prefix를 이용하여 구하는 두 가지 방법이 있어서 간략하게 정리를 해보겠습니다.

String.range(of: "정규 표현식", option: [.regularExpression]) 

위의 String의 타입메서드는 of: 에 해당하는 정규식에 맞는 범위를 리턴합니다. 만약 정규 표현식에 해당하는 것이 없다면 nil을 리턴하므로 if let 구문을 이용하여 범위를 구합니다. 따라서 파일명의 HEAD의 범위는 숫자가 아닌 수가 등장할 때까지의 범위가 됩니다. 

그럼 HEAD부분을 제외하고 숫자가 등장하는 부분의 범위를 구하여 NUMBER를 구할 수 있습니다.

정규 표현식은 구글에 검색하면 자세하게 설명이 되어있습니다. 워낙 양이 많고 방대해서 해당 글에선 설명하지 못하지만 참고하셔서 익혀두시면 문자열을 조작하는 문제에서 더욱 간단한 해결법을 찾을 수 있을 것 같습니다.

String.drop()  -> drop안의 조건식을 만족하는 부분을 삭제하고 나머지 부분을 리턴합니다.
String.prefix() -> 조건식을 만족하는 첫번째 요소를 리턴합니다.

다음은 Range, drop과 prefix를 이용하여 HEAD와 NUMBER를 구해보겠습니다. 

prefix를 이용하여 숫자를 포함하지 않는 부분을 구하여 HEAD를 구하고 , drop을 이용하여 HEAD를 지우고 다시 prefix를 이용하여 NUMBER를 구하는 방식입니다. 

나머지 과정은 똑같이 (HEAD, NUMBER, 입력순서) 이 세 가지를 비교하여 정렬하는 방식으로 해결하면 될 것 같습니다.

---

👍🏻  참고

• https://velog.io/@kerri/Swift-코테-파일명-정렬

[Swift] 프로그래머스(Lv2) - [3차] 파일명 정렬 (2018 KAKAO BLIND RECRUITMENT )

• 프로그래머스 - 님의 풀이