题目:
给你一个整数数组 nums ,返回 nums 中所有 等差子序列 的数目。
如果一个序列中 至少有三个元素 ,并且任意两个相邻元素之差相同,则称该序列为等差序列。
例如,[1, 3, 5, 7, 9]、[7, 7, 7, 7] 和 [3, -1, -5, -9] 都是等差序列。
再例如,[1, 1, 2, 5, 7] 不是等差序列。
数组中的子序列是从数组中删除一些元素(也可能不删除)得到的一个序列。
例如,[2,5,10] 是 [1,2,1,2,4,1,5,10] 的一个子序列。
题目数据保证答案是一个 32-bit 整数。
思路:
动态规划。用n来记录所给数组的长度,f为n个字典的列表。f[i]表示第i个字典。
f[i][d]表示:以nums[i]为尾项,以d为公差的的弱等差子序列的个数。(所谓弱等差子序列:是指等差子序列中至少包含两个数)
分别以nums[j],nums[i]作为弱等差子序列的倒数第二项和最后一项(j<i),且该等差序列的公差为d=nums[i]-nums[j]。
若公差相同,则nums[i]可加到 以nums[j]为尾项的弱等差序列后,构成标准的等差序列(即至少包含三项)。那么状态转移方程可表示为:
f[i][d]=f[i][d]+f[j][d]+1
其中1 表示仅由nums[j]和nums[i]两项组成的弱等差序列。
解答:
class Solution:
def numberOfArithmeticSlices(self, nums: List[int]) -> int:
ans = 0
#用n(nums的数组长度)个字典来统计 公差为d,尾项为nums[i]的弱等差序列的个数
f = [defaultdict(int) for _ in nums]
#print(f)
for i, x in enumerate(nums):
for j in range(i):
d = x - nums[j]
cnt = f[j][d]
#print(j,i,cnt)
ans += cnt
f[i][d] += cnt + 1
#print(i,d,f)
return ans