【LeetCode】413. 等差数列划分

413. 等差数列划分（中等）

1. 思路

   • 由于题目求的是等差数列 ，很自然想到子数组一定满足 nums[i] - nums[i-1] == nums[i-1] -nums[i-2]; 。然而我们对于 dp 数组的定义通常是以 i 结尾，满足某些条件的子数组数量，而等差数组可以在任意一个位置终结，因此此题在最后需要对 dp 数组求和。

2. 代码

   class Solution {
   public:
       int numberOfArithmeticSlices(vector<int>& nums) {
           int n = nums.size();
           if(n <= 2) 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;
               }
           }
           return accumulate(dp.begin(), dp.end(), 0);
       }
   };
   

3. 收获

   • 题解思路很清晰，我反而把这道题想复杂了，此外，我居然还对数组进行排序，导致一开始没办法通过，稀里糊涂。