(力扣—动态规划)区域和检索 - 数组不可变
给定一个整数数组 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 方法。
!特别说明 !
这道题算法没啥的,但是告诉我们一个道理,在力扣上刷题不 要 使 用 全 局 变 量!!!
否则可能会出现执行没错提交出错的情况!
python code
class NumArray:
def __init__(self, nums: list):
if not len(nums): return
self.sums = []
self.sums.append(nums[0])
for i in range(1, len(nums)):
self.sums.append(self.sums[i - 1] + nums[i])
def sumRange(self, i: int, j: int) -> int:
if i == 0:
return self.sums[j]
return self.sums[j] - self.sums[i - 1]
c++ code
class NumArray {
public:
vector<int> sum;
NumArray(vector<int> nums) {
if(!nums.size()) return;
sum.push_back(nums[0]);
for(int i = 1; i < nums.size(); i++)
sum.push_back(nums[i] + sum[i - 1]);
}
int sumRange(int i, int j) {
if(i == 0)
return sum[j];
return sum[j] - sum[i - 1];
}
};