算法概论week14 | leetcode 413. Arithmetic Slices

题目描述

---

分析

这道题比较简单，不用动态规划也能在O(n)的复杂度下得出答案，我主要分析一下用动态规划怎么解决这道题。假设我们已经知道了以A[i]结尾的等差数列的个数（假设为n），那么以A[i + 1]结尾的等差数列的个数也能求出来，如果A[i + 1]与A[i]的差值等于A[i]与A[i - 1]的差值，以A[i + 1]结尾的等差数列的个数就为n + 1，否则为0，而数组 A 中所有为等差数组的子数组个数就等于所有以A中的数为结尾的等差数列个数之和。算法复杂度为O(n)。

---

c++代码

class Solution {
public:
    int numberOfArithmeticSlices(vector<int>& A) {
        int n = A.size();
        if(n < 3) return 0;
        
        vector<int> dp(n, 0);
        if(A[1] - A[0] == A[2] - A[1]) dp[2] = 1;
        int res = dp[2];
        for(int i = 3; i < n; i++) {
            if(A[i] - A[i - 1] == A[i - 1] - A[i - 2]) {
                dp[i] = dp[i - 1] + 1;
            }
            res += dp[i];
        }
        
        return res;
    }
};