【Leetcode】303. 区域和检索 - 数组不可变

QUESTION

easy

题目描述

给定一个整数数组 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

说明

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

---

SOLUTION

此题说明了数组不可变，且又要多次调用 sumRange 方法，故我们可以只保存和的结果。

方法一

• sum[i] 用于保存 0 ~ i 的和
• 多开辟了一个空间节省判断的时间

class NumArray {
public:
    NumArray(const vector<int> & nums) {
        int size = nums.size() + 1;
        sum = new int[size];
        sum[0] = 0;
        for(int i = 1; i < size; i++){
            sum[i] = sum[i - 1] + nums[i - 1];
        }
    }
    
    int sumRange(int i, int j) {
        return sum[j + 1] - sum[i];
    }
private:
    int *sum;
};