题目链接:点击查看
题目描述:
给定一个数组,求这个数组中连续且等差的子数组一共有多少个。 输入输出: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);
}