1. 问题描述:
如果一个数列至少有三个元素,并且任意两个相邻元素之差相同,则称该数列为等差数列。例如,以下数列为等差数列:
1, 3, 5, 7, 9
7, 7, 7, 7
3, -1, -5, -9
以下数列不是等差数列。
1, 1, 2, 5, 7
数组 A 包含 N 个数,且索引从0开始。数组 A 的一个子数组划分为数组 (P, Q),P 与 Q 是整数且满足 0<=P 元素 A[P], A[p + 1], ..., A[Q - 1], A[Q] 是等差的。并且 P + 1 < Q 。
函数要返回数组 A 中所有为等差数组的子数组个数。
示例:
A = [1, 2, 3, 4]
返回: 3, A 中有三个子等差数组: [1, 2, 3], [2, 3, 4] 以及自身 [1, 2, 3, 4]。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/arithmetic-slices
2. 思路分析:
分析题目可以知道等差数列需要满足两个条件:
- 至少包含三个元素
- 相邻两个元素之间的差值是相等的
因为考虑到需要计算相邻两个元素的差值,所以可以使用差分数组来存储相邻两个元素的差,这样我们在原数组中找等差序列等价于在差分数组中找到连续相等的区间,然后计算出区间中等差序列的数目累加起来就是答案了,所以引入等差序列之后很方便的计算每一段连续区间等差序列的个数。
3. 代码如下:
from typing import List
class Solution:
def numberOfArithmeticSlices(self, nums: List[int]) -> int:
n = len(nums)
# 计算相邻元素的差值, 逆序计算是为了防止先计算前面元素的差值对后面元素计算差值有影响
for i in range(n - 1, 0, -1):
nums[i] = nums[i] - nums[i - 1]
i = 1
res = 0
while i < n:
# 这里可以使用双指针算法计算差分数组中连续相等的一段区间
j = i
while j < n and nums[i] == nums[j]:
j += 1
# [i - 1, j - 1]区间的元素都是等差序列, 计算长度大于等于3的数目即可
k = j - i
res += (k - 1) * k // 2
i = j
return res