프로그래머스 - 파일명 정렬(Swift)

프로그래머스 - 파일명 정렬(Swift)

 

 

 문제 설명

 

https://school.programmers.co.kr/learn/courses/30/lessons/17686

프로그래머스

 

 나의 풀이

 

numFlag와 tailFalg변수를 통해서 배열을 쪼갰지만 별로 좋은방법은 아닌 것 같다.

그리고 정렬을 할 때 처음에는 삼항연산자 안에 $0.head < $1.head ? $0.head < $1.head : 나머지로직 이런식으로 작성했는데 이는 첫번째 $0.head < $1.head는 단순히 앞에것이 뒤에것보다 크다면 이기 때문에 이렇게 하면 모든 경우를 구하지않아서 !=으로 바꿔주었다.

 

struct Image {
    let head: String
    let number: String
    let tail: String
}
func solution(_ files:[String]) -> [String] {
    var images: [Image] = []
    for file in files {
        var numFlag = false
        var tailFlag = false
        var head: String = ""
        var num: String = ""
        var tail: String = ""
        
        let arr = Array(file)
        
        for x in arr {
            if !x.isNumber && !numFlag {
                head += String(x)
            } else if x.isNumber && !tailFlag {
                numFlag = true
                num += String(x)
            } else {
                tail += String(x)
                tailFlag = true
            }
        }
        let image = Image(head: head, number: num, tail: tail)
        images.append(image)
        
        
    }
    let b = images.sorted { $0.head.lowercased() != $1.head.lowercased() ? $0.head.lowercased() < $1.head.lowercased() : Int($0.number)! < Int($1.number)! }
    
    return b.map{
        let str = $0.head + String($0.number) + $0.tail
        return str
    }
 
}

 

 

 다른 사람의 풀이

 

prefix(while:) 과 drop(while:)을 사용해서 문제를 풀었다.

prefix(while:)은 조건이 true인 동안의 prefix를 가져오는 것이다. 즉 조건이 false가 되면 prefix가 멈춘다.

drop(while:)은 조건이 true일때 값들을 지우고 false가 되는 순간 그 뒤에 값들은 조건에 걸리지 않고 전체를 리턴한다.

즉

let str = "foo010bar020.zip"라는 문자열이 있을때 foo를 지우고 010bar020.zip이 리턴되는 것이다.

 

그리고 sort하는 부분에서는 head의 이름도 같고 number의 이름도 같을 때는 false를 줬는데 이유는 입력값 순서대로 리턴을 하면 되기 때문이다. 그래서 숫자를 바꾸지 않도록 했다.

 

struct Word {
    let head: String
    let number: String
    let index: Int
}

func makeWord(_ str: String, _ idx: Int) -> Word {
    let head = str.prefix{ !$0.isNumber }.lowercased()
    let num = str.drop{ !$0.isNumber }.prefix{ $0.isNumber }
    return Word(head: head, number: String(num), index: idx)
}

func AsortB(_ lhs: Word, _ rhs: Word) -> Bool {
    let aNum = Int(lhs.number)!, bNum = Int(rhs.number)!
    if lhs.head != rhs.head { return lhs.head < rhs.head }
    if aNum != bNum { return aNum < bNum }
    return false
}


func solution(_ files:[String]) -> [String] {
    var words: [Word] = []
    
    for (i, file) in files.enumerated() {
        words.append(makeWord(file, i))
    }
    
    words.sort { AsortB($0, $1) }
    let result = words.map{ files[$0.index] }
    
    return result
 
}