给定一个整数数组,计算其等差子数组的个数(至少有算个元素)。如对于数组[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; }