PrefixSum_Leetcode_303. Range Sum Query - Immutable [swift]

https://leetcode.com/problems/range-sum-query-immutable/description/

 

정답코드

아무 생각 없이 nums만 저장하고 method에서 더해서 반환했더니

35ms가 나왔다. (0ms가 나와야함)

PrefixSum 문제라는걸 상기하고 다시 풀어서 0ms로 해결

class NumArray {

    var nums: [Int]
    var sums: [Int]

    init(_ nums: [Int]) {
        self.nums = nums    
        var cur = 0
        var sums = [Int](repeating: 0, count: nums.count)
        for i in 0..<nums.count {
            cur += nums[i]
            sums[i] = cur
        }
        self.sums = sums
    }
    func sumRange(_ left: Int, _ right: Int) -> Int {
        if left > 0 {
            return sums[right] - sums[left - 1]
        } else if right > 0 {
            return sums[right]
        } else { 
            return sums[0] 
        }
        
    }

    // func sumRange(_ left: Int, _ right: Int) -> Int {
    //     let ranged = nums[left...right]
    //     return ranged.reduce(0,{$0 + $1})
    // }
}