interviewalgo
, , ,
Source
Edit
History

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

Introduction

Question: 303. 区域和检索 - 数组不可变

给定一个整数数组  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< code>
  • 最多调用 104 次 sumRange 方法

解法一

Analysis

用前缀和即可。

sumRange(left, right) = sum[0:right+1] - sum[0:left]

由于-10^5 <= nums[i] <= 10^5，所以没有溢出问题。

Implement

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
class NumArray {
public:
    NumArray(vector<int>& nums): sum(nums.size() + 1) {
        sum[0] = 0;
        int i = 0;
        for(auto &v: nums) {
            sum[i+1] = sum[i] + v;
            i++;
        }
    }
    
    int sumRange(int left, int right) {
        return sum[right+1] - sum[left];
    }
private:
    vector<int> sum;
};