方法一:
从数列头开始,将以该元素为开头的奇数长度子数组遍历求和,所以需要遍历该数列、计算以该元素为开头的数列长、遍历该奇数长数组求和。
public int sumOddLengthSubarrays(int[] arr) {
int sum = 0;
//遍历该数列
for (int i = 0; i < arr.length; i++) {
//计算以该元素为开头的数列长
for (int j = 1; j <= arr.length; j += 2) {
if (i + j - 1 < arr.length){
//遍历该奇数长数组求和
for (int k = 0; k < j; k++) {
sum += arr[i + k];
}
}
}
}
return sum;
}
方法二:
用前缀和,即将当前下标前所有元素加起来作为一个新数组存起来,这样在求方法一的第三步就不用遍历求和,直接利用新数组求解。(注意下标越界问题)
public int sumOddLengthSubarrays(int[] arr) {
int sum = 0;
int[] pre = new int[arr.length];
for (int i = 0; i < arr.length; i++) {
pre[i] = arr[i] + (i - 1 < 0 ? 0 : pre[i - 1]);
}
for (int i = 0; i < arr.length; i++) {
for (int j = 1; j <= arr.length; j += 2) {
if (i + j - 1 < arr.length){
sum += pre[i + j - 1] - (i - 1 < 0 ? 0 : pre[i - 1]);
}
}
}
return sum;
}
方法三:
主要思想是计算当前元素可能出现在多少个子数组当中,即当前元素会对最后的结果付出多少贡献数值。
主要问题就是求当前数可能出现在多少个子数组中。
当当前元素左侧为奇数个元素且右侧也有奇数个元素时,会有一种奇数组合。
当当前元素左右任意一侧或同时为偶数时,会有一种组合。
所以需要求出左侧奇数、偶数的可能,右侧奇数、偶数的可能,利用上述桂策进行排列组合(即数量相乘即可),再乘当前元素数值即可。
public int sumOddLengthSubarrays(int[] arr) {
int sum = 0;
for (int i = 0; i < arr.length; i++) {
int leftOdd = (i + 1) / 2;
int rightOdd = (arr.length - i) / 2;
int leftEven = i / 2 + 1;
int rightEven = (arr.length - i - 1) / 2 + 1;
sum += arr[i] * (leftOdd * rightOdd + leftEven * rightEven);
}
return sum;
}
(注:0也算偶数,所以算偶数都要加1)
偶数和奇数相比就是除以二前数值需要减一,可以这么想,既然计算组合为奇数的个数:
当总数为奇数时,java计算除2相当于自动将总数减1求解,假如长度为3,数值3本身也会组成一次奇数,但除法只会得出1,所以算奇数就需要额外加1,协助最外侧的数字完成在计算奇数组合时判断正确。