一、题目
如果一个数列 至少有三个元素 ,并且任意两个相邻元素之差相同,则称该数列为等差数列。
例如,[1,3,5,7,9]、[7,7,7,7] 和 [3,-1,-5,-9] 都是等差数列。
给你一个整数数组 nums ,返回数组 nums 中所有为等差数组的 子数组 个数。
子数组 是数组中的一个连续序列。
点击查看原题
难度级别: 中等
二、思路
1)公差+计数
所求子数组为原数组的连续序列,如果有公差为d的子数组长度为m
,那么可以划分为满足题意子数组个数为:
(
m
−
3
+
1
)
+
(
m
−
4
+
1
)
+
.
.
.
+
(
m
−
m
+
1
)
=
(
m
−
1
)
∗
(
m
−
2
)
/
2
(m-3+1)+(m-4+1)+...+(m-m+1)=(m-1)*(m-2)/2
(m−3+1)+(m−4+1)+...+(m−m+1)=(m−1)∗(m−2)/2
只需要遍历整个数组并记录公差和最长子数组长度,再按上式计算累加即可(只计算连续子数组长度大于等于3
的)
三、代码
1)公差+计数
class Solution {
public int numberOfArithmeticSlices(int[] nums) {
if (nums.length < 3) {
return 0;
}
int ans = 0;
int cnt = 2;
int margin = nums[1] - nums[0];
for (int i = 2; i < nums.length; i++) {
int t = nums[i] - nums[i-1];
if (t != margin) {
margin = t;
ans += (cnt-1)*(cnt-2)/2;
cnt = 2;
} else {
cnt++;
}
}
ans += (cnt-1)*(cnt-2)/2;
return ans;
}
}
时间复杂度为O(n)
,空间复杂度为O(1)