LeetCode 413. 等差数列划分

最新推荐文章于 2024-06-11 16:51:29 发布

_almost__

最新推荐文章于 2024-06-11 16:51:29 发布

阅读量164

点赞数

分类专栏：刷题

本文链接：https://blog.csdn.net/almost_afei/article/details/119568925

版权

刷题专栏收录该内容

293 篇文章 0 订阅

订阅专栏

这篇文章探讨了如何使用动态规划算法解决数组中算术序列的问题。作者给出了一个名为Solution的类，通过迭代检查相邻元素的差值来确定连续子数组是否构成算术序列，并计算满足条件的序列总数。关键步骤包括长度更新和计数累积。

摘要由CSDN通过智能技术生成

难度：中等。
标签：数组，动态规划。

正确解法：

class Solution {
public:
    int numberOfArithmeticSlices(vector<int>& nums) {
        int n = nums.size();
        int ans = 0;
        int len = 0;
        for(int i = 2; i < n; ++i){
            if(nums[i] - nums[i - 1] == nums[i - 1] - nums[i - 2]){
                if(len == 0)len = 3;
                else len += 1;
            }
            else if(len > 0){
                ans += (double)(len - 1) / 2.0 * (len - 2);
                len = 0;
            }
        }
        if(len > 0){
            ans += (double)(len - 1) / 2.0 * (len - 2);
        }
        return ans;
    }
};