Programmers: K번째 수

단순무식하게 commands 배열 순회하면서 요구 사항대로 맞춰서 코딩했다. 예전에 풀었던 파이썬 코드를 보면 엄청 간결하게 해놨는데, 스위프트로는 그렇게 구현하려면 subscript를 구현하면 될 거 같은데 일단은 패스.

 

문제

https://programmers.co.kr/learn/courses/30/lessons/42748?language=swift

 

내가 작성한 코드

Swift

import Foundation

func solution(_ array:[Int], _ commands:[[Int]]) -> [Int] {
    var answer = [Int]()
    for command in commands {
        let i = command[0]-1
        let j = command[1]-1
        let k = command[2]-1
        var arr = [Int]()
        for a in i...j {
            arr.append(array[a])
        }
        answer.append(arr.sorted()[k])
    }
    return answer
}