题目地址:
https://www.lintcode.com/problem/interval-sum/description
给定一个数组,要求设计一个数据结构,实现区间和的查询。
最方便的实现区间和查询的数据结构是前缀和数组。需要注意的是,前缀和数组 S S S的长度应该开成原数组的长度多一个, S [ i ] S[i] S[i]表示原数组 A A A的前 i i i个数之和,那么 S [ 0 ] = 0 S[0]=0 S[0]=0就表示原数组的前 0 0 0个数之和,也就是 0 0 0。 ∑ i = j k A [ i ] = S [ k + 1 ] − S [ j ] \sum_{i=j}^{k}A[i]=S[k+1]-S[j] i=j∑kA[i]=S[k+1]−S[j]区间和可以在 O ( 1 ) O(1) O(1)的时间内得到。代码如下:
import java.util.ArrayList;
import java.util.List;
public class Solution {
/**
* @param A: An integer list
* @param queries: An query list
* @return: The result list
*/
public List<Long> intervalSum(int[] A, List<Interval> queries) {
// write your code here
long[] preSum = new long[A.length + 1];
// 计算前缀和数组
for (int i = 0; i < A.length; i++) {
preSum[i + 1] = preSum[i] + A[i];
}
List<Long> res = new ArrayList<>();
for (Interval query : queries) {
long end = preSum[query.end + 1], start = preSum[query.start];
res.add(end - start);
}
return res;
}
}
class Interval {
int start, end;
Interval(int start, int end) {
this.start = start;
this.end = end;
}
}
预处理时间复杂度 O ( n ) O(n) O(n),空间 O ( n ) O(n) O(n),查询时空复杂度 O ( 1 ) O(1) O(1)。