프로그래머스 - K번째수 / Swift

문제 링크 : https://programmers.co.kr/learn/courses/30/lessons/42748 

코딩테스트 연습 - K번째수

 

 

풀이

commands 배열을 돌며 array 배열을 잘라 임시 배열 tmp에 넣는다.

tmp 배열을 정렬한 뒤 k번째 수를 정답 배열 ans에 넣어 리턴한다.

 

코드 

import Foundation

func solution(_ array:[Int], _ commands:[[Int]]) -> [Int] {
    var ans: [Int] = []
    
    for i in commands{
        var tmp: [Int] = []
        
        for j in i[0]...i[1]{
            tmp.append(array[j-1])
        }
        tmp.sort()
        ans.append(tmp[i[2] - 1])
    }
    
    return ans
}