[내일배움캠프] 데일리 루틴(iOS_3회차) - K번째수

 

이미지를 클릭하면 코딩테스트 페이지로 이동합니다

 

 

내 풀이

import Foundation

func solution(_ array:[Int], _ commands:[[Int]]) -> [Int] {
    return commands.map({(key) in
        return array[(key[0]-1)...(key[1]-1)].sorted()[key[2]-1]
    })
}

 

 

commands.map({ (key) in ... })

     commands 배열을 순회하며 각 명령어에 대해 클로저 실행
     → 클로저 내부에서는 각 명령어(key)에 따라 배열을 슬라이스하고 정렬하여 결과를 반환

array[(key[0]-1)...(key[1]-1)].sorted()[key[2]-1]

     각 명령어에 대해 배열을 슬라이스한 후 정렬
     → 배열의 인덱스는 0부터 시작하므로, 명령어에서 주어진 인덱스를 1씩 감소시켜야 함
     → 따라서 key[0]-1과 key[1]-1로 슬라이스를 수행하고, 정렬된 배열에서 key[2]-1 인덱스의 값 반환