区域和检索 - 数组不可变

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

实现 NumArray 类：

  • NumArray(int[] nums) 使用数组 nums 初始化对象

  • int sumRange(int i, int j) 返回数组 nums 从索引 i 到 j（i ≤ j）范围内元素的总和，包含 i、j 两点（也就是 sum(nums[i], nums[i + 1], … , nums[j])）

  • 示例：

    输入：
["NumArray", "sumRange", "sumRange", "sumRange"]
[[[-2, 0, 3, -5, 2, -1]], [0, 2], [2, 5], [0, 5]]
输出：
[null, 1, -1, -3]

    解释：

    • NumArray numArray = new NumArray([-2, 0, 3, -5, 2, -1]);
    • numArray.sumRange(0, 2); // return 1 ((-2) + 0 + 3)
    • numArray.sumRange(2, 5); // return -1 (3 + (-5) + 2 + (-1))
    • numArray.sumRange(0, 5); // return -3 ((-2) + 0 + 3 + (-5) + 2 + (-1))

提示：

  • 0 <= nums.length <= 104
  • -105 <= nums[i] <= 105
  • 0 <= i <= j < nums.length
  • 最多调用 104 次 sumRange 方法

Solution

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
#include <iostream>
#include <vector>
using namespace std;

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


int main() {
    vector<int> input;
    vector<int> result;

    input = {-2, 0, 3, -5, 2, -1};
    std::cout << NumArray(input).sumRange(0, 2) << std::endl; // 1
    std::cout << NumArray(input).sumRange(2, 5) << std::endl; // -1
    std::cout << NumArray(input).sumRange(0, 5) << std::endl; // -3
}