python leetcode 413. Arithmetic Slices

最新推荐文章于 2020-01-30 18:13:41 发布

Neekity

最新推荐文章于 2020-01-30 18:13:41 发布

阅读量509

点赞数

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

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

版权

python 同时被 2 个专栏收录

263 篇文章 0 订阅

订阅专栏

leetcode

221 篇文章 2 订阅

订阅专栏

class Solution(object):
    def numberOfArithmeticSlices(self, A):
        """
        :type A: List[int]
        :rtype: int
        """
        res=0
        if len(A)<3:return 0
        pre=A[1]-A[0]
        index=0
        for i in range(2,len(A)):
            if pre!=A[i]-A[i-1]:
                if i-index>=3:res+=self.count(i-index)
                pre=A[i]-A[i-1]
                index=i-1 
        if i-index+1>=3:res+=self.count(i-index+1)
        return res 
    def count(self,n):
        return sum(i for i in range(1,n-1))