[프로그래머스]테이블 해시 함수(Swift)

brick·2023년 3월 7일
0

코테

목록 보기
43/53
import Foundation

func solution(_ data:[[Int]], _ col:Int, _ row_begin:Int, _ row_end:Int) -> Int {
    let data = data.sorted(by: {
        if $0[col-1] == $1[col-1] { return $0[0] > $1[0] }
        else { return $0[col-1] < $1[col-1] }
    })
    
    var arr: [Int] = []
    for i in row_begin-1...row_end-1 {
        var sum: Int = 0
        for n in data[i] {
            sum += n%(i+1)
        }
        arr.append(sum)
    }
    
    return arr.reduce(0, ^)
}

0개의 댓글