个人博客主页:部依的技术小窝 (zioplyyc.fun)
前缀和
问题 :
给定一个长度问 N 的整数数组和两个数 L、R,返回该数组当中 L 到 R 的数字和
解决方案一:
创建一个 N x N 的二维数组进行打表,记录从从 0-N 之间所有的区间和。
public class RangeSum1 {
private int[] arr;
private int[][] db;
public RangeSum1(int[] arr) {
this.arr = arr;
db = new int[arr.length][arr.length];
for (int i = 0; i < arr.length; i++) {
db[i][i] = arr[i];
for (int j = i + 1; j < arr.length; j++) {
db[i][j] = db[i][j - 1] + arr[j];
}
}
}
public int rangeSum(int L, int R) {
return db[L][R];
}
}
优点:
- 适合查询次数非常多的情况,在查询非常多的情况下,方法一比方法二少一次减法过程
缺点:
- 前期建立二位数组的时候计算量达到 N x N / 2 的量级
解决方案二:
创建一个长度为 n 的一位数组(H),记录从 0 到 i 的前缀和,使用的时候如果 L 是 0 则直接返回 H[R] 。如果 L 不是 0 则返回 H[R] - H[L-1] 的值
public class RangeSum2 {
private int[] preSum;
public RangeSum2(int[] arr) {
int N = arr.length;
preSum = new int[N];
preSum[0] = arr[0];
for (int i = 1; i < arr.length; i++) {
preSum[i] = preSum[i - 1] + arr[i];
}
}
public int rangeSum(int[] arr, int L, int R) {
return L == 0 ? preSum[R] : preSum[R] - preSum[L - 1];
}
}
优点
- 适合查询次数不是特别多的情况
- 建立数组只需要遍历以便数组就行,少的查询次数运算量小
缺点
- 数据量过大,查询效率没有方法一高