【Leetcode 431】等差数列划分

等差数列划分

题目

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

• 例如，[1,3,5,7,9]、[7,7,7,7] 和 [3,-1,-5,-9] 都是等差数列。

• 给你一个整数数组nums，返回数组 nums 中所有为等差数组的子数组个数。

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

• 来源：力扣（LeetCode）
  链接：https://leetcode-cn.com/problems/arithmetic-slices
  著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题解

规律性思考

• [1,2,3]含有1个等差数列，
• [1,2,3,4]含有3个等差数列，
• [1,2,3,4,5]含有6个等差数列，
• [1,2,3,4,5,6]含有10个等差数列。
• 依次类推，在num[i,..,j-1]为等差数列的基础上，加入num[j]，若能构成等差数列，则数列数的增量+1。

原理性思考

差分+计数

题目本身需要枚举等差数列的最后两项计算不同的差值，即对于所有d = num[i-1] - num[i]，设置指针j从i - 2的位置逆序扫描，

• 当num[j]-num[j+1] = d时，说明num[j…i]至少包含一组等差数列，数列+1。
• 否则跳出循环（因为指针是逆序扫描，更小的 j 也无法作为等差数列的首个位置），计算下一个差值下等差数列的增量。

优化

上述算法存在重叠子问题：
即当num[i] - nums[i+1] == nums[i-1] - nums[i]时，j会遍历到同样的位置，考虑使用dp算法，设dp[i]为差值为d=num[i]-num[i+1]时等差数列的增量。

• 如果 num[i] - nums[i+1] == nums[i-1] - nums[i]，答案增加的次数与d=num[i-1]-num[i]相同，并且额外多出了nums[i-1],nums[i],nums[i+1]这一个等差数列，因此有： dp[i]=dp[i-1]+1;
• 否则，dp[i] = 0;

空间优化：
又因为dp[i]只与dp[i-1]的状态有关，所以只需记录dp[i-1]即可。

代码实现（c++）

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