leetcode每日一题—413.等差数列划分

题目：

如果一个数列 至少有三个元素 ，并且任意两个相邻元素之差相同，则称该数列为等差数列。

例如，[1,3,5,7,9]、[7,7,7,7] 和 [3,-1,-5,-9] 都是等差数列。
给你一个整数数组 nums ，返回数组 nums 中所有为等差数组的 子数组 个数。

子数组 是数组中的一个连续序列。

思路：
滑动窗口，满足等差条件，则累加结果，扩大窗口；否则滑动窗口，更新公差。

解答：

class Solution:
    def numberOfArithmeticSlices(self, nums: List[int]) -> int:
        n=len(nums)
        if n<3:
            return 0
        #找最长子数组
        left,right=0,1
        dif=nums[right]-nums[left]
        res=0
        while right<n:
            #print(left,right)
            #不满足等差条件，滑动窗口，更新公差
            if nums[right]-nums[right-1]!=dif:              
                left=right-1
                dif=nums[right]-nums[left]
            #满足等差条件，累加结果，扩大窗口
            else:
                #累加 以right为右端点的长度大于等于3的子数组数量
                if right-left>1:
                    res+=(right-left-1)
                    #print(left,right,res)    
                right+=1     
        return res