给定一个整数数组 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 方法。
思路:这道题采用动态规划来做,维护一个一维数组dp,其中dp[i]表示一直累加到第i位的和,那么计算第[i,j](包含第i位,包含第j位)的值就等于:
dp[j]-dp[i-1]
这里要重点考虑第0位的问题。
参考代码:
class NumArray {
public:
NumArray(vector<int> nums) {
dp=nums;
for (int i = 0; i < nums.size(); i++) {
if (i == 0) {
dp[i] = nums[0];
}
else {
dp[i] += dp[i - 1];
}
}
}
int sumRange(int i, int j) {
return i == 0 ? dp[j] : (dp[j] - dp[i - 1]);
}
private:
vector<int> dp;
};