LeetCode Arithmetic Slices(动态规划)

最新推荐文章于 2025-01-31 17:09:35 发布

原创最新推荐文章于 2025-01-31 17:09:35 发布 · 554 阅读

0 ·

CC 4.0 BY-SA版权

OJ 同时被 2 个专栏收录

986 篇文章

订阅专栏

算法设计与分析

321 篇文章

订阅专栏

本文介绍了一种使用动态规划算法解决寻找等差子数组问题的方法。通过维护一个计数器来跟踪当前连续等差元素的数量，并利用这些信息计算总的等差子数组数量。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意：给出一个数组，求其连续子数组（最小长度为3，为等差数列）的个数

思路：用动态规划，用 dp(i)表示到数组索引i时的满足条件的个数,其中i>=3,有dp(i) = dp(i- 1) + cnt(这个表示连续满足两个之差相等的计数)

代码如下：

public class Solution
{
    public int numberOfArithmeticSlices(int[] A)
    {
        if (A.length < 3) return 0;
        int len = A.length;
        int ans = 0;
        int cnt = 0;
        for (int i = 2; i < len; i++)
        {
            if (A[i - 2] - A[i - 1] == A[i - 1] - A[i])  cnt++;
            else cnt = 0;

            ans += cnt;
        }

        return ans;
    }
}