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的升级版,那是我第一道做不出来放弃的题目,下篇讲。