[Swift 알고리즘] 프로그래머스 Lv1 k번째수

programmers.co.kr/learn/courses/30/lessons/42748

나의 풀이

import Foundation

func solution(_ array:[Int], _ commands:[[Int]]) -> [Int] {
 
    var result: [Int] = []
    
    commands.forEach { test in
        let index = array.index(test[0]-1, offsetBy: test[1]-test[0]+1)
        let a = Array(array[test[0]-1..<index]).sorted()[test[2]-1]
        result.append(a)
    }
    return result
}

 

다른 사람의 풀이

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]
        })

    }

 

배울 점

1. 따로 마지막 index를 찾아서 쓰는게 아니라 저렇게 쓰면 더 좋음

2. map을 활용하자