给定一个整数数组,计算其等差子数组的个数(至少有算个元素)。如对于数组[1, 2, 3, 4],其等差子数组有[1, 2, 3], [2, 3, 4] 和 [1, 2, 3, 4]。
分析:只需遍历一次,把等差数列的长度分别计算出来。如果长度为3,其中包含的子数组个数为1,长度为4,包含 2 + 1 个,长度为5,包含 3 + 2 + 1 个,依此类推。
public int numberOfArithmeticSlices(int[] A) {
int index = 0;
int sum = 0;
while(index < A.length) {
int l = 2;
while((index + l < A.length) && A[index + l] - A[index + l - 1] == A[index + 1] - A[index]) l++;
if(l >= 3)
for(int i = l - 2; i > 0; i--)
sum += i;
index += l - 1; //A[index + l]- A[index + l - 1]与前面的差不相等,下一步从A[index + l - 1]开始计算
}
return sum;
}
847

被折叠的 条评论
为什么被折叠?



