LeetCode413等差数列划分（动态规划）

题目链接：leetcode413

思路：动态规划，$dp[i]$表示以i位置的数为结尾中形成等差数列的个数的集合。则每次如果以$A[i]、A[i-1]、A[i-2]$形成一个等差数列，则这个A[i]有可能由和之前i-1位置为结尾形成的所有等差数列分别拼接而成，所以计算集合的值时要加上$dp[i-1]$。最后统计所有结尾的等差数量sum即可。

class Solution {
public:
    int numberOfArithmeticSlices(vector<int>& A) {
        int n = A.size();
        if(n < 3) return 0;
        vector<int> dp(n); //dp[i]表示以i位置的数为结尾中形成等差数列的个数
        dp[0] = dp[1] = 0;
        int sum = 0;
        for(int i=2; i<n; i++) {
            if (A[i - 1] - A[i - 2] == A[i] - A[i - 1]) {
                dp[i] = dp[i - 1] + 1; //当前合法的等差数列长度>=3(以i为结尾等于3的等差数列个数为1,大于3的个数为dp[i - 1])
            } else {
                dp[i] = 0;
            }
            sum += dp[i]; //每个位置为结尾的都加上
        }
        return sum;
    }
};