前缀和技巧用于计算一个区间内元素之和。
比如:给定一个数组nums, 元素为[0,1,2,3,4], 计算任意区间内元素的和。
我们可以使用循环来逐一累加区间内的元素,达到求和的目的。但是这样的话,每次求和都需要时间复杂度O(n);
而使用前缀和技巧可以将时间复杂度降到O(1);
前缀和是数组该位置之前的元素之和,这里我们使用前缀和核心是求出前缀和数组,然后在前缀和数组之中寻找答案。
new一个新的数组perSum出来,然后把nums的前缀和数组求出来,然后直接在perSum中寻找答案。
上代码:
class NumArray{
//前缀和数组
private int[] perSum;
public NumArray(int[] nums){
perSum = int[nums.length + 1];
perSum[0] = 0;
//填充perNums数组
for(int i=1; i < perNums.length; i++){
perSum[i] = perSum[i-1] + nums[i-1];
}
}
//求left到right下标区间之和
public int sumRange(int left, int right){
return perSum[right+1] - perSum[left];
}
}
我们实现这样一个类,然后在构造方法中直接生成前缀和数组,那么我们在求区间和的时候只需要调用sumRange数组就可以直接通过O(1)的时间复杂度就可以得出结论。
缺点就是多占用了一个前缀和数组的空间,空间复杂度相对高一些。