출처: 프로그래머스 코딩 테스트 연습

문제 설명

세 차례의 코딩 테스트와 두 차례의 면접이라는 기나긴 블라인드 공채를 무사히 통과해 카카오에 입사한 무지는 파일 저장소 서버 관리를 맡게 되었다.

저장소 서버에는 프로그램의 과거 버전을 모두 담고 있어, 이름 순으로 정렬된 파일 목록은 보기가 불편했다. 파일을 이름 순으로 정렬하면 나중에 만들어진 ver-10.zip이 ver-9.zip보다 먼저 표시되기 때문이다.

버전 번호 외에도 숫자가 포함된 파일 목록은 여러 면에서 관리하기 불편했다. 예컨대 파일 목록이 ["img12.png", "img10.png", "img2.png", "img1.png"]일 경우, 일반적인 정렬은 ["img1.png", "img10.png", "img12.png", "img2.png"] 순이 되지만, 숫자 순으로 정렬된 ["img1.png", "img2.png", "img10.png", img12.png"] 순이 훨씬 자연스럽다.

무지는 단순한 문자 코드 순이 아닌, 파일명에 포함된 숫자를 반영한 정렬 기능을 저장소 관리 프로그램에 구현하기로 했다.

소스 파일 저장소에 저장된 파일명은 100 글자 이내로, 영문 대소문자, 숫자, 공백(" "), 마침표("."), 빼기 부호("-")만으로 이루어져 있다. 파일명은 영문자로 시작하며, 숫자를 하나 이상 포함하고 있다.

파일명은 크게 HEAD, NUMBER, TAIL의 세 부분으로 구성된다.

• HEAD는 숫자가 아닌 문자로 이루어져 있으며, 최소한 한 글자 이상이다.

• NUMBER는 한 글자에서 최대 다섯 글자 사이의 연속된 숫자로 이루어져 있으며, 앞쪽에 0이 올 수 있다. 0부터 99999 사이의 숫자로, 00000이나 0101 등도 가능하다.

• TAIL은 그 나머지 부분으로, 여기에는 숫자가 다시 나타날 수도 있으며, 아무 글자도 없을 수 있다.

파일명	HEAD	NUMBER	TAIL
foo9.txt	foo	9	.txt
foo010bar020.zip	foo	010	bar020.zip
F-15	F	15	(빈 문자열)

파일명을 세 부분으로 나눈 후, 다음 기준에 따라 파일명을 정렬한다.

• 파일명은 우선 HEAD 부분을 기준으로 사전 순으로 정렬한다. 이때, 문자열 비교 시 대소문자 구분을 하지 않는다. MUZI와 muzi, MuZi는 정렬 시에 같은 순서로 취급된다.

• 파일명의 HEAD 부분이 대소문자 차이 외에는 같을 경우, NUMBER의 숫자 순으로 정렬한다. 9 < 10 < 0011 < 012 < 13 < 014 순으로 정렬된다. 숫자 앞의 0은 무시되며, 012와 12는 정렬 시에 같은 같은 값으로 처리된다.

• 두 파일의 HEAD 부분과, NUMBER의 숫자도 같을 경우, 원래 입력에 주어진 순서를 유지한다. MUZI01.zip과 muzi1.png가 입력으로 들어오면, 정렬 후에도 입력 시 주어진 두 파일의 순서가 바뀌어서는 안 된다.

무지를 도와 파일명 정렬 프로그램을 구현하라.

출력 형식

위 기준에 따라 정렬된 배열을 출력한다.

입출력 예

s	n	result
"AB"	1	"BC"

입출력 예제

입력: ["img12.png", "img10.png", "img02.png", "img1.png", "IMG01.GIF", "img2.JPG"]
출력: ["img1.png", "IMG01.GIF", "img02.png", "img2.JPG", "img10.png", "img12.png"]

입력: ["F-5 Freedom Fighter", "B-50 Superfortress", "A-10 Thunderbolt II", "F-14 Tomcat"]
출력: ["A-10 Thunderbolt II", "B-50 Superfortress", "F-5 Freedom Fighter", "F-14 Tomcat"]

내가 제출한 코드

import Foundation

// 문제에선 tail 사용 x
func solution(_ files:[String]) -> [String] {
    let answer = files.sorted(by: { // 파일 정렬
        let firstFile = getHeadAndNumber($0) //(head: HEAD, number: NUMBER)
        let secondFile = getHeadAndNumber($1)
        
        // head가 다른 경우 head로 정렬
        if firstFile.head.lowercased() != secondFile.head.lowercased() { 
            return firstFile.head.lowercased() < secondFile.head.lowercased()
        }
        // head가 같은 경우 NUMBER로 정렬
        return firstFile.number < secondFile.number 
    })
    return answer
}

// 파일에서 HEAD와 NUMBER 반환
func getHeadAndNumber(_ file :String) -> (head: String, number :Int) {
    var copyFile: String = file
    var head: String = ""
    var number: String = ""
    
    // head 구함
    for (i,chr) in copyFile.enumerated() {
        if chr.isNumber { // 숫자를 입력받은 경우 이전까지는 Head가 됨
            let headIndex = copyFile.index(copyFile.startIndex, offsetBy: i)
            head = String(copyFile[..<headIndex]) // head 추출
            copyFile.removeSubrange(..<headIndex) // file에서 head 제거
            break
        }
    }
    
    if let num = Int(copyFile) { // tail인 빈 문자열인 경우 파일의 문자열을 전부 반환
        return (head,num)
    }

    // NUMBER 계산
    for (i, chr) in copyFile.enumerated() {
        if !chr.isNumber { // 숫자가 아닌 문자가 오면 그전까지 넘버
            let numberIndex = copyFile.index(copyFile.startIndex, offsetBy: i)
            number = String(copyFile[..<numberIndex])
            break
        }
    }
    return (head, Int(number)!)
}

코드 설명

이번 문제는 정렬 문제이다. 파일명이 숫자, 문자 등등..으로 구성되어 있고 아래와 같이 세 부분으로 구성된다.

1. HEAD
2. NUMBER
3. TAIL

그래서 처음 떠올린 생각은 파일명을 세 부분으로 구분하는 것이었다. 그러나 문제를 잘 읽어보면 정렬은 앞의 2가지인 HEAD와 NUMBER만 사용하는 것을 알 수 있다. 즉 정렬에는 2가지만 사용된다. 그러므로 파일명에서 HEAD와 NUMBER의 값만 다루고 TAIL 부분은 굳이 다룰 필요가 없다.

그러면 코드를 살펴볼 차례인데, 우선 파일명을 분리하는 코드를 보고, 다음으로 정렬을 수행하는 코드의 순서로 설명한다.

파일명에서HEAD 부분을 분리하는 코드다.

// 파일에서 HEAD와 NUMBER 반환
func getHeadAndNumber(_ file :String) -> (head: String, number :Int) {
    var copyFile: String = file
    var head: String = ""
    var number: String = ""
    
    // head 구함
    for (i,chr) in copyFile.enumerated() {
        if chr.isNumber { // 숫자를 입력받은 경우 이전까지는 Head가 됨
            let headIndex = copyFile.index(copyFile.startIndex, offsetBy: i)
            head = String(copyFile[..<headIndex]) // head 추출
            copyFile.removeSubrange(..<headIndex) // file에서 head 제거
            break
        }
    }

파일명의 HEAD는 최소 1개의 문자로 구성되어 있다. NUMBER도 최소 1개의 숫자를 가지며, 위 2가지를 제외한 나머지는 TAIL이며, 빈 문자열일 수도 있다. 그러므로 HEAD는 NUMBER의 앞자리까지의 문자열을 의미한다. 이를 찾기 위해 파일명을 숫자가 나올 때까지 한 문자씩 반복하여 읽는다. 숫자가 나온다면 이전까지의 문자들을 합한 문자열이 HEAD가 된다. 이를 head 프로퍼티에 할당한다.

우리는 HEAD를 얻었으므로 NUMBER를 얻어야 한다. 얻기 전에 파일명에서 HEAD 를 제거하였다. 그러므로 NUMBER와 TAIL만 문자열에 남아있다. TAIL이 빈문자열 인 경우 남은 파일명이 NUMBER이 되므로 그대로 반환한다. 아래 코드는 이를 확인한다.

if let num = Int(copyFile) { // tail인 빈 문자열인 경우 파일의 문자열을 전부 반환
    return (head,num)
}

다음은 NUMBER를 구하는 코드이다.

 // NUMBER 계산
    for (i, chr) in copyFile.enumerated() {
        if !chr.isNumber { // 숫자가 아닌 문자가 오면 그전까지 넘버
            let numberIndex = copyFile.index(copyFile.startIndex, offsetBy: i)
            number = String(copyFile[..<numberIndex])
            break
        }
    }
    return (head, Int(number)!)

TAIL이 빈 문자열이 아니라면 우리는 NUMBER와 TAIL을 분리해야 한다. 분리하는 방법은 간단하다. NUMBER는 연속되는 최대 5개의 숫자로 이루어져 있으므로 숫자가 아닌 문자를 읽을 때까지 반복문을 순회한다. 만약 숫자가 아닌 다른 문자를 찾았다면 이전까지 읽은 문자열을 number 프로퍼티에 할당한다. 그리고 함수의 제일 마지막에는 찾은 NUMBER와 TAIL을 반환한다.

마지막으로 아래 코드는 정렬을 수행하는 코드다.

let answer = files.sorted(by: { // 파일 정렬
        let firstFile = getHeadAndNumber($0) //(head: HEAD, number: NUMBER)
        let secondFile = getHeadAndNumber($1)
        
        // head가 다른 경우 head로 정렬
        if firstFile.head.lowercased() != secondFile.head.lowercased() { 
            return firstFile.head.lowercased() < secondFile.head.lowercased()
        }
        // head가 같은 경우 NUMBER로 정렬
        return firstFile.number < secondFile.number 
    })
    return answer

이제 우리는 파일명을 정렬하기 위해 정렬에 사용되는 2개의 부분을 얻었다. 이 2가지 부분을 가지고 정렬을 수행하면 된다. 우선 입력으로 들어온 파일을 정렬하기 위해 sorted(by:) 메소드를 사용한다. 정렬을 수행하는 방법은 HEAD의 값이 다르다면 HEAD로 비교하고, 같다면 NUMBER의 값으로 비교한다. 이때 HEAD는 대소문자가 상관없다는 점과 NUMBER의 값을 문자열로 비교하는 것이 아닌 Int 타입으로 비교하는 것을 유의하자.