[LeetCode]413.Arithmetic Slices

原题地址

题意：

如果一组数包含至少3个元素，并且任意两个连续元素之差都相等，则称该序列为等差序列。
给定一个以0为起始下标的数组A，包含N个数字。数组的切片是指任意满足0 <= P < Q < N的整数对 (P, Q)。
数组A的切片 (P, Q) 是等差数列，如果满足下列条件：
A[P], A[p + 1], …, A[Q - 1], A[Q]是等差数列。特别的，P + 1 < Q。
函数应当返回数组A的等差数列切片的个数

大神算法

时间 O(n)

public int numberOfArithmeticSlices(int[] A) {
    int curr = 0, sum = 0;
    for (int i=2; i<A.length; i++)
        if (A[i]-A[i-1] == A[i-1]-A[i-2]) {
            curr += 1;
            sum += curr;
        } else {
            curr = 0;
        }
    return sum;
}

思路：

如果是不连续的情况，比如 ： 1231234
很明显 sum （最终的值） = sum(123) + sum(1234) 两段等差片段相互独立

那么问题简化为求一个连续等差的片段有多少个？

1，2，3，4，，，n 的等差数列的片段数 =
以1结尾的等差数列片段数 +
以2结尾的等差数列片段数 +
以3结尾的等差数列片段数 +
以4结尾的等差数列片段数 +
……
以n结尾的等差序列片段数

问题简化为求以X结尾的等差片段有多少个？

记 以x结尾的等差片段（就是一定要包含x） 为 D(x)，有：

D(x) = D(x-1) + 1

首先以D(x-1)结尾的片段，其后再加一个等差的数，必然还是等差的
然后以 X-1 结尾的，(X-2,X-1), 再加X，构成了新的等差片段 (X-2,X-1,X),所以要再加1