给定一个整数数组 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 方法。
思路:如果要是每次都遍历数组求子序列和的话,因为会调用很多次肯定会超时。所以我们设置数组re,re[i+1]表示该数组前i个元素的和。每次调用sumRange的时候,直接返回re[j+1]-re[i]即可。在re前面塞进去一个0防止越界。
class NumArray {
public:
NumArray(vector<int>& nums) {
for(auto x:nums){
sum.push_back(x+sum.back());
}
}
int sumRange(int i, int j) {
return sum[j+1]-sum[i];
}
private:
vector<int>sum{0};
};