等差数列划分（双指针or差分）

Posted 2021-09-04 秦枫-_-

滑动窗口优化：双指针O(N)，i-idx-1之间是一个等差数列，那么求它的等差子数组即可，最小的等差子数组长度为3，数量为len-3+1，最长的等差子数组长度为len，数量为1，那么他子数组的数量也是一个等差为1的等差数列，套用等差数列求和公式，最后i不必从最开始查起，直接跳转到idx-2处 以idx-2为首查等差数列长度即可

class Solution {
    public int numberOfArithmeticSlices(int[] nums) {
     if(nums.length<=2)return 0;
     int ans=0;
     for(int i=0;i<nums.length-2;i++){
        int idx=i+2;
        while(idx<nums.length){
            if(nums[idx]-nums[idx-1]==nums[idx-1]-nums[idx-2]){
                idx++;
            }
            else break;
        }
        int len=idx-i;
        int a3=len-3+1,a1=1;
        ans+=(a3+a1)*a3/2;
        i=idx-2;
     }
     return ans;
    }
}

curans每次记录的是以idx-1结尾的等差子数组个数

class Solution {
    public int numberOfArithmeticSlices(int[] nums) {
     if(nums.length<=2)return 0;
     int ans=0;
     int curans=0;
     for(int i=0;i<nums.length-2;i++){
        int idx=i+2;
        if(nums[idx]-nums[idx-1]==nums[idx-1]-nums[idx-2]){
                curans++;      
        }
        else curans=0;
         ans+=curans;   
        
     }
     return ans;
    }
}