8-10每日一题：等差数列划分

最新推荐文章于 2022-08-15 16:58:28 发布

原创最新推荐文章于 2022-08-15 16:58:28 发布 · 116 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法

每日一题专栏收录该内容

144 篇文章

订阅专栏

本文介绍了如何利用等差数列的特性，通过差分操作预处理数组，计算连续差值相等的子数组个数。Solution类中定义了get_cnt函数和numberOfArithmeticSlices方法，用于解决此问题。核心在于遍历数组，寻找符合条件的子数组并累加计数。

等差数列划分

差分

预处理查分数组，因为等差数组是连续的子数组，找连续的长度大于等于2的差相等的子数组的个数，累加起来返回

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

        for (int i = 1; i < n; i ++ )
            d[i + 1] = nums[i] - nums[i - 1];

        int ans = 0;
        for (int i = 2; i < n; i ++ )
        {
            int j = i;
            while (j + 1 <= n && d[j + 1] == d[i]) j ++ ;
            ans += get_cnt(i ,j + 1);
            i = j;
        }
        return ans;
    }
};