🌕🌗 413. 等差数列划分

难度: 🌕🌗

问题描述

---

解法

class Solution {
    public int numberOfArithmeticSlices(int[] nums) {
        // 思路：
        // dp[i] - 以 i 为结尾下标的等差数列的种类数
        // 注意题意的子数组是指，数组中的连续序列
        int len = nums.length;
        if(len < 3) {
            return 0;
        }
        // len >= 3
        int res = 0;
        int[] dp = new int[len];
        for(int i = 2; i < len; i ++) {
            // 判断 [i] 能否与 [i - 1] & [i - 2] 形成等差数列
            if(nums[i] - nums[i - 1] == nums[i - 1] - nums[i - 2]) {
                // 满足连续的子序列
                // {[i - 2], [i - 1], [i]} 可以单独组成一个等差序列
                dp[i] = 1;
                // [i] 可以与 dp[i - 1] 构成连续的等差序列
                dp[i] += dp[i - 1];
                res += dp[i];
            } 
        }
        return res;
    }
}

---

输出