[Swift] Stride

💻 프로그래머스 : 과일 장수 - 문제 링크

---

첫 풀이

import Foundation

func solution(_ k: Int, _ m: Int, _ score: [Int]) -> Int {
    var newScore = score
    var cost = 0
    newScore = newScore.filter { $0 <= k }.sorted(by: >)

    let boxCount = newScore.count / m

    for i in 0..<boxCount {
        let boxMinValue = newScore[(i * m) + (m - 1)]
        cost += boxMinValue * m
    }

    return cost
}

다른사람 풀이

import Foundation

func solution(_ k:Int, _ m:Int, _ score:[Int]) -> Int {
    let s = score.sorted(by: >)
    return stride(from: m-1, to: score.count, by: m)
        .reduce(0) { $0 + s[$1] * m }
}

---

stride()

일정한 간격을 두고 시퀀스를 생성하는 데 사용
시작값부터 끝값까지 일정하게 증가하거나 감소

stride() 함수 기본 구조

1. stride(from: 시작값, to: 끝값(포함X), by: 간격)
2. stride(from: 시작값, through: 끝값(포함O), by: 간격)

stride() 함수 활용하여 배열 생성

let array1 = Array(stride(from: 1, to: 10, by: 2))
print(array1)  // 출력: [1, 3, 5, 7, 9]

python list comprehension같은게 혹시 없나 하고 처음에 찾았었는데, 비슷한건 찾은 것 같다. 더 편한 여러 메서드들을 찾고 익히고 싶다.
앞으로 다양한 방법으로 활용할 수 있을 것 같다.