题目
给你一个正整数数组 arr ,请你计算所有可能的奇数长度子数组的和。
子数组 定义为原数组中的一个连续子序列。
请你返回 arr 中 所有奇数长度子数组的和 。
输入:arr = [1,4,2,5,3]
输出:58
解释:所有奇数长度子数组和它们的和为:
[1] = 1
[4] = 4
[2] = 2
[5] = 5
[3] = 3
[1,4,2] = 7
[4,2,5] = 11
[2,5,3] = 10
[1,4,2,5,3] = 15
我们将所有值求和得到 1 + 4 + 2 + 5 + 3 + 7 + 11 + 10 + 15 = 58
方法一:暴力
自己的方法:三重循环!(时间复杂度高了点)
一重:控制子数组的奇数长度
二重:准备遍历子数组,控制好条件
三重:遍历left到right(right取不到) 的元素,计算sum。
public int sumOddLengthSubarrays(int[] arr) {
int sum = 0;
int len = 1;
while (len <= arr.length) {
int left = 0;
int right = len;
while (left < arr.length && right <= arr.length) {
for (int i = left; i < right; i++) {
sum += arr[i];
}
left++;
right++;
}
len += 2;
}
return sum;
}
方法二:前缀和
我一开始没有明白为什么这道题可以用前缀和,但是要记住,求一个区间范围的数组的和,是可以用前缀和解决的!
思路:
先求出arr数组的前缀和。
然后设置两个指针,一个i (在for循环中,可以充当一下),一个是end。因为是奇数长度的区间内求和,设置len来控制数组的长度,我们还要控制求和的对应元素:
res+=sum[end]-sum[i]
其中的 end - i==len的。
因为前缀和可以求出一段区间内的所有元素的和,所以用后面的sum[end]减去前面的sum[i]可以求出某个区间的总和,这样我们就可以解决这道题了。
前缀和变通!!!
计算arr数组的前缀和,并让sum[0]=0,这样可以方便计算,保证end - i==len,同时避免res+=sum[end]-sum[i]中的 i 越界。
如果是这种写法:
private int[] demo(int[] src) {
int[] result = new int[src.length];
result[0]=src[0];
for (int i = 1; i < result.length; i++) {
result[i] = result[i - 1] + src[i];
}
return result;
}
保证不了end - i==len。
所以应该这样写:
int[] sum = new int[arr.length + 1];
sum[0] = 0;
for (int i = 1; i < sum.length; i++) {
sum[i] = sum[i - 1] + arr[i - 1];
}
完整代码
public int sumOddLengthSubarrays2(int[] arr) {
int[] sum = new int[arr.length + 1];
sum[0] = 0;
for (int i = 1; i < sum.length; i++) {
sum[i] = sum[i - 1] + arr[i - 1];
}
int res = 0;
int len = 1;
while (len < sum.length) {
for (int i = 0; i < sum.length; i++) {
int end = len + i;
if (end >= sum.length) {
break;
}
res += sum[end] - sum[i];
}
len += 2;
}
return res;
}
如果不是很明白的小伙伴,可以在草稿纸上面演练一下。
方法三:(研究中)
还有一种方法我没有懂,段位不够,以后再来看看。
参考
反思
动手动脑,继续加油!
end.