题目描述:
给定一个整数数组 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])
)
方法一:
class NumArray {
public:
vector<int> num;
NumArray(vector<int>& nums) {
num = nums;
}
int sumRange(int left, int right) {
int sum = 0;
for(int i=left ; i<=right ; i++)
{
sum+=num[i];
}
return sum;
}
};
/**
* Your NumArray object will be instantiated and called as such:
* NumArray* obj = new NumArray(nums);
* int param_1 = obj->sumRange(left,right);
*/
最简单的思路,初始化时用数组num来保存传入nums,然后每次调用sumRange时都计算一遍对应的元素和。
看完官方的题解之后,才知道了用前缀和的方法:初始化时,记录下nums数组中每个元素的前i项和,则在计算某两个下标之间的元素之和时,直接相减即可。
方法二:
class NumArray {
public:
vector<int> sum;
NumArray(vector<int>& nums) {
int n = nums.size();
sum.resize(n+1);
for(int i=0 ; i<n ; i++)
{
sum[i+1] = sum[i] + nums[i];
}
}
int sumRange(int left, int right) {
return sum[right+1]-sum[left];
}
};
/**
* Your NumArray object will be instantiated and called as such:
* NumArray* obj = new NumArray(nums);
* int param_1 = obj->sumRange(left,right);
*/
sum[i+1]的意思是nums数组中的前i项和,而sum[0]等于0,之所以使用i+1,就是为了确保在sumRange中计算时不用额外考虑i=0的特殊情况。
时间复杂度:初始化 O(n),每次检索 O(1),其中 n 是数组 nums 的长度。
初始化需要遍历数组 nums 计算前缀和,时间复杂度是 O(n)。
每次检索只需要得到两个下标处的前缀和,然后计算差值,时间复杂度是 O(1)。
空间复杂度:O(n),其中 n 是数组 nums 的长度。需要创建一个长度为 n+1 的前缀和数组。
两种方法的提交对比:
可以发现,第二次提交的方法二所用的执行时间比方法一要少的多