题目
给定一个整数数组 nums,求出数组从索引 i 到 j(i ≤ j)范围内元素的总和,包含 i、j 两点。
实现 NumArray 类:
NumArray(int[] nums) 使用数组 nums 初始化对象
int sumRange(int i, int j) 返回数组 nums 从索引 i 到 j(i ≤ j)范围内元素的总和,包含 i、j 两点(也就是 sum(nums[i], nums[i + 1], … , nums[j]))
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/range-sum-query-immutable
方法一、遍历
每次都从i遍历到j计算答案
class NumArray {
private:
vector<int> nums;
public:
NumArray(vector<int>& nums) {
this->nums = nums;
}
int sumRange(int i, int j) {
int res = 0;
for(int k = i; k <= j; k++)
res+=nums[k];
return res;
}
};
方法二、分块和
将nums以block_size的大小分块计算和,并提前存储到一个数组中
class NumArray {
private:
int block_size;
vector<int> nums;
vector<int> sums;
public:
NumArray(vector<int>& nums) {
this->nums = nums;
block_size = 100;
int sum = 0;
int cnt = 0;
for(int i = 0; i < nums.size(); ++i)
{
cnt++;
sum += nums[i];
if(cnt==100)
{
cnt = 0;
sums.push_back(sum);
sum = 0;
}
}
}
int sumRange(int i, int j) {
int res = 0;
int k = i;
while(k<=j)
{
// [0,99] [100,199],[200,299]
if(k%block_size==0 && k+block_size-1 <= j)
{
res += sums[k/block_size];
k += block_size;
}else
{
res += nums[k];
++k;
}
}
return res;
}
};
方法三、前缀和
- 前缀和:前缀和是一个数组的某项下标之前(包括此项元素)的所有数组元素的和。
- 一维情况下: presum[i] = presum[i-1]+num[i]
本题求的是i-j的和,如果用presum[j]-presum[i] 得到的是i+1到j的和,固用presum[i+1] = presum[i]+nums[i]
class NumArray {
public:
vector<int> presum;
NumArray(vector<int>& nums) {
int n = nums.size();
presum.resize(n + 1);
for (int i = 0; i < n; i++) {
presum[i + 1] = presum[i] + nums[i];
}
}
int sumRange(int i, int j) {
return presum[j + 1] - presum[i];
}
};