[프로그래머스] K번째수

<문제>

---

<전략짜기>

문제에서 하라는 대로만 하면 되는..? 문제 같다. 크게 생각해야 할 부분은 없을 것 같았고, 한가지 주의해야 할 부분은 commands에서 각 요소(배열)의 값들은 "n번째" 이지 인덱스가 "n"임을 나타내는 것이 아니므로, Out Of Index Error를 피하기 위해 commands의 각 요소들은 인덱스로 활용하기 위해 모두 -1을 처리한 후에 사용 해야 한다는 점이다.

 

<코드>

import Foundation

func solution(_ array:[Int], _ commands:[[Int]]) -> [Int] {
    var result: [Int] = []
    for command in commands {
        var orderArray = array[command[0]-1...command[1]-1].sorted()
        result.append(orderArray[command[2]-1])
    }
    return result
}

 

<Lesson Learned>

오랜만에 알고리즘을 다시 시작하는 점(대학생때 이후로 안했으니.. 6년만? 인거같다 ㅠ), Swift를 공부하기 위해 시작했다는 점 때문에 Level 1 난이도 문제에서 골라풀고 있다..하하 😂 그중에서도 이 문제는 쉬웠던 문제 같다. 그러나.... 역시 다른사람의 풀이를 봤을때 이럴 수도 있구나.. 싶은건 있다 ㅠㅠ 이번에도 map을 사용했다면 result라는 배열도 필요없었을 것 같고, 코드도 간결했을 것 같다. 아래처럼 한번 돌려봤는데, 잘 된다 😁😁 난이도 올려서 문제 풀려면 map, filter에 빨리 익숙해질 필요가 있다. 👊

import Foundation

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

 

 

출처 : 프로그래머스 K번째수