【leetcode】303 区域和检索 - 数组不可变（前缀和）

题目链接：https://leetcode-cn.com/problems/range-sum-query-immutable/

题目描述

给定一个整数数组 nums，求出数组从索引 i 到 j (i ≤ j) 范围内元素的总和，包含 i, j 两点。

示例：

给定 nums = [-2, 0, 3, -5, 2, -1]，求和函数为 sumRange()

sumRange(0, 2) -> 1
sumRange(2, 5) -> -1
sumRange(0, 5) -> -3

说明:

1. 你可以假设数组不可变。
2. 会多次调用 sumRange 方法。

思路

前缀和数组
时间复杂度：O（1）。指sumRange函数的时间复杂度。
空间复杂度：O（n）

class NumArray {
    vector<int> sums;   // sums[i]表示前i项和
public:
    NumArray(vector<int>& nums) {
        int sum = 0;
        sums.push_back(sum);    // 前0项和为0
        for(int i = 0; i < nums.size(); ++i){
            sum+= nums[i];
            sums.push_back(sum);
        }
    }
    
    int sumRange(int i, int j) {
        return sums[j+1] - sums[i]; // 注意前j项和为sums[j+1]
    }
};

/**
 * Your NumArray object will be instantiated and called as such:
 * NumArray* obj = new NumArray(nums);
 * int param_1 = obj->sumRange(i,j);
 */