Leetcode 413. 等差数列划分 通俗详解-滑动窗口和动态规划

---

题目

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

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

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

示例1：

输入：nums = [1,2,3,4]
输出：3
解释：nums 中有三个子等差数组：[1, 2, 3]、[2, 3, 4] 和 [1,2,3,4] 自身。

示例2：

输入：nums = [1]
输出：0

提示：

1 <= nums.length <= 5000
1000 <= nums[i] <= 1000

思路

首先看到题目是数组，一般会想到双指针，再进一步会想到滑动窗口或者动态规划，这道题两种方法都可以做，第二种更快些。

滑动窗口思路： 等差数组最小长度是3，由此可先定义i和j，分别指向第一个位置和第三个位置；再定义一个位置变量dist，表示两个数之间的差值。如下dist = nums[i + 1] - nums[i];可用第二个和第一个位置的差值初始化dist，然后从j开始遍历while(j < n && nums[j] - nums[j - 1] == dist)此时是第三个第二个之间的差值，一直循环到不满足条件。即滑动窗口中，j滑动到最右端，此时滑动左端（即i++），如此循环得到最终结果。

滑动窗口

class Solution {
public:
    int numberOfArithmeticSlices(vector<int>& nums) {
        //滑动窗口，
        int n = nums.size();
        if(n <= 2) return 0;
        int ans = 0,j,dist;
        for(int i = 0; i < n - 2; i++) {
            j = i + 2;
            dist = nums[i + 1] - nums[i];
            while(j < n && nums[j] - nums[j - 1] == dist) {
                j++;
                ans++;
            }
        }
        return ans;
    }
};

动态规划

最重要的思路是：等差数组的子数组也必定是等差数组。假设有[1,2,3,4,5]下面代码从第三个位置开始判断：nums[i] - nums[i-1] == nums[i-1] - nums[i-2] ?可知道[1,2,3]是等差数组，此时++l得到l=1，然后是[1,2,3,4]，++l得到l=2，因为有等差数组：[1, 2, 3]、[2, 3, 4] 和 [1,2,3,4] 自身，结果为3。再进一步[1,2,3,4,5]，++l得到l=3，结果为6。变量l记录着上一个位置的等差数组的数量，这其实也是他的状态方程。
（这是评论区中一个大佬的代码，简洁美观）

class Solution {
public:
    int numberOfArithmeticSlices(vector<int>& nums) {
        int n = nums.size(), l = 0, ans = 0;
        for(int i = 2; i < n; i++)  
            nums[i] - nums[i-1] == nums[i-1] - nums[i-2] ? ans += ++l : l = 0;
        return ans;
    }
};

大佬的代码是经过优化的，空间复杂度是O(1)，那么我把它反优化是这个样子，更加通俗易懂：
dp[i]：此位置的等差数组个数
状态方程：dp[i] = dp[i - 1] + 1;，很容易想到，即假设[0,1,2,3]，dp[2]里有[0,1,2]，dp[3]里有[1,2,3]、[0,1,2,3]。

class Solution {
public:
    int numberOfArithmeticSlices(vector<int>& nums) {
        int n = nums.size(),ans = 0;
        vector<int>dp(n,0);
        for(int i = 2; i < n; i++) {
            if(nums[i] - nums[i-1] == nums[i-1] - nums[i-2]) {
                dp[i] = dp[i - 1] + 1;
                ans += dp[i];
            }
        }
        return ans;
    }
};

413. 等差数列划分