LeetCode 413. Arithmetic Slices

题目

Return the number of arithmetic slices in the array A.
数组A中的递增子数组的总数。

输入：A = [1, 2, 3, 4]
输出：3

思路

这道题发现官方是把它当成dp的思路，我是把它用滑动窗口的想法做的，窗口标记为[start, end），表明这个窗口中是当前完整递增数组。

就拿题目例子来说，4加入进去的时候总数增加了多少呢？ 这其实取决于当前这个递增数组的长度。这里增加的是[1,2,3,4] 和 [2,3,4] 这两个子数组，也就是增加了1开头和2开头的两个数组。

因为一个递增数组的最小长度是3，也就是说新增元素 Ai 新生成的数组开头必须在 i - 2 及其左侧。假如我们记录当前完全递增数组的开头为start (例子中start=0) , 第i个元素若和前面形成连续递增，则增加的子数组开头可以取 [start，i-2]，其个数是 i-1-start。

当这个元素不和前面形成连续递增时，此时递增数组停止增长，新窗口开启。

int numberOfArithmeticSlices(vector<int>& A) {
        int number = 0;
        if(A.size()<3)
            return 0;
        int start = 0;
        int end = 1;
        for(int i=2;i<A.size();i++){
        	//如果当前i元素符合前面的递增，窗口扩展，总数增加
            if(A[i]-A[i-1] == A[i-1]-A[i-2]){
                int add_slices_num = i-start-1;
                end ++; 
                number += add_slices_num;
            }
            //如果不符合等差递增，开启新窗口
            else{
                start = end;
                end += 1;
            }
        }
        return number;
    }

官方的题解是把它当做dp问题， 我们往常会把dp[i]记录成到 i 为止的结果，但这道题不能设置成子数组总和，不然 i ->i+1没法状态转换。 这里应该记录的是以 i 结尾的子数组个数。这样一来，状态转移：

$dp[i+1]=dp[i]+1$

理解为每一个 i 结尾子数组都可以产生 一个插入i+1的新数组，同时，还创造了一个i+1结尾的长度为3的新数组。

要求的和就是把每一个dp[i]加起来。

状态压缩后的代码如下（dp[i+1]之和前一个有关，只要记录一个值）：

 int numberOfArithmeticSlices(vector<int>& A) {
        int number = 0;
        if(A.size()<3)
            return 0;
        int pre = 0;
        for(int i=2;i<A.size();i++){
            if(A[i]-A[i-1] == A[i-1]-A[i-2]){
                pre += 1;
                number += pre;
            }           
            else{
               pre = 0;
            }
        }
        return number;
    }