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

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

题目

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

题解

利用前缀和
dp[i]表明数组前i+1个数的和，我们要计算nums[i]~nums[j]时，只需要dp[j]-dp[i-1]，若i==0，则直接返回nums[j]

代码

	vector<int> dp;
    NumArray(vector<int>& nums) {
        if(nums.empty())    return;
        int n=nums.size();
        dp = vector(n,0);
        dp[0] = nums[0];
        for(int i=1; i<n; i++)
            dp[i] = dp[i-1]+nums[i]; 
    }
    
    int sumRange(int i, int j) {
        if(i == 0)  return dp[j];
        return dp[j] - dp[i-1] ;
    }

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/range-sum-query-immutable
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。