leetCode 413 等差数列划分（一维dp）

题目链接：点击查看

题目描述：

给定一个数组，求这个数组中连续且等差的子数组一共有多少个。

 

输入输出：
 

A = [1, 2, 3, 4]

返回: 3, A 中有三个子等差数组: [1, 2, 3], [2, 3, 4] 以及自身 [1, 2, 3, 4]。

题目分析：

这道题略微特殊，因为要求是等差数列，可以很自然的想到子数组必定满足 num[i] - num[i-1] = num[i-1] - num[i-2]。然而由于我们对于 dp 数组的定义通常为以 i 结尾的，满足某些条件的子数组数量，而等差子数组可以在任意一个位置终结，因此此题在最后

 

需要对 dp 数组求和。

 

代码：

int numberOfArithmeticSlices(vector<int>&nums)
{
   int n=nums.size();
   if(n<3)
   return 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;//对于第 i 个元素，判断这个元素跟前一个元素的差值是否和等差数列中的差值相等。
		                //如果相等，那么新区间中等差数列的个数即为 1+dp[i-1]。
	 }
   }
   return accumulate(dp.begin(),dp.end(),0);	
}