Leecode413等差数列划分
题目
- 如果一个数列 至少有三个元素 ,并且任意两个相邻元素之差相同,则称该数列为等差数列。
- 例如,[1,3,5,7,9]、[7,7,7,7] 和 [3,-1,-5,-9] 都是等差数列。
给你一个整数数组 nums ,返回数组 nums 中所有为等差数组的 子数组 个数。 - 子数组 是数组中的一个连续序列
示例:
-
输入:nums = [1,2,3,4] 输出:3 解释:nums 中有三个子等差数组:[1, 2, 3]、[2, 3, 4] 和 [1,2,3,4] 自身。
-
输入:nums = [1] 输出:0
算法思路
-
考虑一个比较直观的做法:
我们枚举等差数列的最后两项nums[i−1] 以及 nums[i],那么等差数列的公差 dd 即为 nums[i−1]−nums[i];
随后我们使用一个指针 jj 从 i - 2i−2 开始逆序地遍历数组的前缀部分nums[0…i−2]:
如果nums[j] -nums[j + 1] = d,那么说明[i]nums[j],⋯,nums[i] 组成了一个长度至少为 33 的等差数列ÿ