如果一个数列 至少有三个元素 ,并且任意两个相邻元素之差相同,则称该数列为等差数列。
例如,[1,3,5,7,9]、[7,7,7,7] 和 [3,-1,-5,-9] 都是等差数列。
给你一个整数数组 nums ,返回数组 nums 中所有为等差数组的 子数组 个数。
子数组 是数组中的一个连续序列。
思路:
从数组的第一个元素开始遍历,判断以第一个元素开始的所有子数组是否为等差数列,首先计算出前两个元素的差,如果第三个元素和第二个元素的差相等,则等差数组的个数加一,遇到不相等的情况直接退出当前元素的循环,从下一个元素开始遍历
代码:
public int numberOfArithmeticSlices(int[] nums) {
if (nums.length <= 2) {
return 0;
}
int ans = 0;
for (int i = 0; i < nums.length -1; i ++) {
int t = 0;
int d = nums[i] - nums[i+1];
for (int j = i +2; j < nums.length;j++) {
if (nums[j - 1] - nums[j] == d) {
t++;
} else {
break;
}
}
ans += t;
}
return ans;
}