leetcode每日一题—446.等差数列划分II-子序列

题目：
给你一个整数数组 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