LeetCode解题报告 413. Arithmetic Slices [medium]

题目描述

A sequence of number is called arithmetic if it consists of at least three elements and if the difference between any two consecutive elements is the same.

For example, these are arithmetic sequence:

1, 3, 5, 7, 9
7, 7, 7, 7
3, -1, -5, -9

The following sequence is not arithmetic.

1, 1, 2, 5, 7

A zero-indexed array A consisting of N numbers is given. A slice of that array is any pair of integers (P, Q) such that 0 <= P < Q < N.

A slice (P, Q) of array A is called arithmetic if the sequence:
A[P], A[p + 1], ..., A[Q - 1], A[Q] is arithmetic. In particular, this means that P + 1 < Q.

The function should return the number of arithmetic slices in the array A. 

Example:

A = [1, 2, 3, 4]

return: 3, for 3 arithmetic slices in A: [1, 2, 3], [2, 3, 4] and [1, 2, 3, 4] itself.

解题思路

题目虽然写的很长，简而言之就是一句话，得到一个数列的所有可组成等差数列的子串的数目。

用一列数来寻找规律，比如1234678。

起始数列定位123，而后一个一个加入。

可以想象的到，在已经是最小个数的3位等差数列数组123的基础上，再加上一个满足等差的这个差值的4，新数列1234的所有等差子串数就是在原来123的等差字数列的基础上，加上4和其他数字组成的新等差子串数，显然就是4加上之前的2、3、4...个数，本数列只有4个数字，因此4可以和之前的2个、3个数组成等差子数列，新增2个，共1+2＝3个。再增加数字5，5可以和之前的2、3、4个数组成新等差子串，共3+3=6个。

当加入7之后，不符合之前的等差的差值，新增数为0；

直到加入9之后，又构成一个新的等差，类似12345的分析。

设置一个raise变量，表示每次在原有基础上新增的等差子数列。

时间复杂度分析

扫描一遍数列即可，时间复杂度为O(N)。

代码如下：

class Solution {
public:
    int numberOfArithmeticSlices(vector<int>& A) {
        int result=0;
        int raise=0;
        for (int i=2; i<A.size(); i++) {
            if (A[i]-A[i-1]==A[i-1]-A[i-2]) {
                raise++;
                result=result+raise;
            }
            else
                raise=0;
        }
    return result;
    }
};