前缀和概念
前缀和,即前缀和数组,给定一个已有数组,我们统计当前下标i之前所有数的和,放进前缀和数组中。看例子:
已知数组:int[] nums = {1, 2, 3, 4, 5, 6};
其对应的前缀和数组:
int[] nums = {1, 2, 3, 4, 5, 6};
int length = nums.length;
// 该前缀和数组长度和原数组一样
int[] preSum1 = new int[length];
preSum1[0] = nums[0];
// 这里的i是 前缀和数组的下标
for (int i = 1; i < length; i++) {
preSum1[i] = preSum1[i-1] + nums[i];
}
// 打印结果:
{1,3,6,10,15,21}
不过一般在做算法题时,前缀和数组的长度 比 原数组多一位
从前缀和数组下标来看就是前0个数的和、前1个数的和…前n个数的和
也就是,第一个数就是0,第二个数的下标为1,所以这个位置就是代表原数组前1个数的和,…,最后一个数就是原数组所有数的和。
这种前缀和数组的生成方式有两种:一个是以前缀和数组的下标为基准,一个是以原数组下表为基准,关键就是在于边界的处理,至于采用哪一种都行,只要明白前缀和数组和原数组的下标之间的关系即可。
int[] nums = {1, 2, 3, 4, 5, 6};
int length = nums.length;
int[] preSum2 = new int[length+1];
// 这里的i是 前缀和数组的下标,所以访问原数组时,下标是i-1,preSum2[0]就是默认值0
for (int i = 1; i <= length; i++) {
preSum2[i] = preSum2[i-1] + nums[i-1];
}
System.out.println("===============");
int[] preSum3 = new int[length+1];
// 这里的i是 原数组的下标
for (int i = 0; i < length; i++) {
preSum3[i+1] = preSum3[i] + nums[i];
}
// 打印结果:
{0, 1 ,3 ,6 ,10 ,15 ,21}
===============
{0, 1 ,3 ,6 ,10 ,15 ,21}
对应题
前缀和:
剑指 Offer II 008. 和大于等于 target 的最短子数组