413. Arithmetic Slices - Medium

Description

（原题描述太罗嗦了，我转换一下语言）
算出一个数组中的等差序列个数，其中找到的序列为连续子串。
Example:
input: 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.

Analyze

目标等差序列都成片（slices）地呆在数组中，对于每片等差序列，只要知道长度，则计算总个数是容易的。即，(n-1)(n-2)/2。
而要找出所有的片，只需往右扫一遍，扫到的这个数对得上我当前片的差，那它就成功的追加到这个片后面了，否则的话要另起新片。一边扫一边计数即可。

Code

int numberOfArithmeticSlices(vector<int>& A) {
    if(A.size() < 3) return 0;
    int num = 2, count = 0, diff = A[1] - A[0];
    for(int i = 2; i < A.size(); ++i) {
        if(A[i]-A[i-1] == diff) num++;
        else {
            if(num > 2) count += (num-1)*(num-2)/2;
            diff = A[i]-A[i-1];
            num = 2;
        }
    }
    //对这个数组恰好是一整片的特殊处理
    if(num > 2) count += (num-1)*(num-2)/2;
    return count;
}

Summary

这道题解法可以说很直观了。
大水题来的，它有一道hard的升级版，那是我第一道做不出来放弃的题目，下篇讲。