[programmers/Swift] K번째수

programmers.co.kr - 코딩테스트연습 - Lv.1 - 정렬 - K번째수

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

Constraints :

• array의 길이는 1 이상 100 이하입니다.
• array의 각 원소는 1 이상 100 이하입니다.
• commands의 길이는 1 이상 50 이하입니다.
• commands의 각 원소는 길이가 3입니다.

 

Solution.swift :

//
//  Created by Yongwoo Marco on 2022/05/08.
//  Copyright © 2022 Yongwoo Marco Kim. All rights reserved.
//

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

print(solution([1, 5, 2, 6, 3, 7, 4], [[2, 5, 3], [4, 4, 1], [1, 7, 3]])) // [5, 6, 3]

How I tried this :
ClosedRange를 이용해서 배열의 부분을 구하고 정렬시킨뒤 k번째 수의 접근

문제에 관한 모든 저작권 : https://programmers.co.kr/