给定一个整数数组 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 方法。
解答:
采用概率分布函数的思想,算出sum[i],那么sum(i,j) = sum[j] - sum[i-1]
class NumArray {
public:
vector<int> dp;
NumArray(vector<int> nums) {
int sum = 0;
for(int i=0;i<nums.size();i++){
sum += nums[i];
dp.push_back(sum);
}
}
int sumRange(int i, int j) {
if(i>=1) return dp[j]-dp[i-1];
else return dp[j];
}
};