LeetCode : 413. 等差数列划分

题目描述：

 

解题思路：

  滑动窗口。窗口左边从下标0开始，右边从下标2开始，判断当前右下标的值能否与之前构成等差数列。

1.   是。结果加1（最左到当前构成等差数列），再加上 最左+1 与 当前-1 之间新增等差数列数。例如：1,2,3,4,5，当窗口到5时，除了加上1到5这个等差数列，还要加上2,3,4新增的个数（即1个）。1,2,3,4,5,6,窗口到6，加上2,3,4,5新增的个数（有3个，但2,3,4在前面算过了，所以2个·，即 i - left - 3）
2.   否。结果加上以左+1 ，左+ 2 ...到 当前-1的等差数列数。修改等差数列差distance，以及窗口left。例如1,2,3,4,5,7,8，当右窗口到7时，不符合，结果加上2到5，3到5即2个等差数列（即i - left - 3），将左窗口指到5，distance为7-5=2。
3.  右边继续滑动。滑到最后判断是否需要加上对应个数（i - left - 3，相当于一次否的情况）。

public int numberOfArithmeticSlices(int[] A) {
        int len = A.length;
        // 边界值
        if (len < 3) {
            return 0;
        }

        // 设初值
        int left = 0;
        int res = 0;
        int distance = A[1] - A[0];

        // 右窗口滑动
        for(int i = 2;i < len;i++) {
            if(A[i] - A[i-1] == distance) {
                // 加上新增个数
                res += 1;
                if(i - left -1 >= 3) {
                    res += i - left - 3;
                }
            } else {
                // 相当于加上移动左窗口新增个数
                if (i - left > 3) {
                    res += i - left - 3;
                }
                // 滑动左窗口 改变公差
                left = i - 1;
                distance = A[i] - A[i-1];
            }
        }
        // 加上移动左窗口新增个数
        if(len - left > 3) {
            res += len - left -3;
        }
        return res;
    }