python leetcode 446. Arithmetic Slices II - Subsequence

最新推荐文章于 2022-07-21 15:55:11 发布

Neekity

最新推荐文章于 2022-07-21 15:55:11 发布

阅读量593

点赞数

分类专栏： leetcode python 文章标签： leetcode python

本文链接：https://blog.csdn.net/Neekity/article/details/84713622

版权

python 同时被 2 个专栏收录

263 篇文章 0 订阅

订阅专栏

leetcode

221 篇文章 2 订阅

订阅专栏

由于求的是有多少种很容易想到用DP做难点在于如何设置及更新DP
假设dp[i][j]是在A[:i+1]差值为j的所有长度为2的数组数量
譬如[2,4,6,8]
i=3时 dp[:3]=[{},{2: 1},{4: 1, 2: 2}]
j=2时 d=A[i]-A[j]=2 d在dp[j]中那么更新dp[i][d]+=dp[j][d] (2,([2,4],[4,6]))=3 ([2,4],[4,6],[6,8]) 相应的total结果+=dp[j][d] ([2,4,6],[4,6,8])

class Solution:
    def numberOfArithmeticSlices(self, A):
        """
        :type A: List[int]
        :rtype: int
        """ 
        n = len(A)
        total=0
        dp=[{} for _ in range(n)]
        for i in range(n):
            for j in range(i):
                d =  A[i]-A[j]
                if d in dp[i]:
                    dp[i][d]+=1
                else:
                    dp[i][d]=1 
                if d in dp[j]:
                    dp[i][d]+=dp[j][d]
                    total+=dp[j][d]
        return total