小猴编程周赛C++ | 2023年春第8周T2等差数列

学习C++从娃娃抓起！记录下在学而思小猴编程学习过程中的题目，记录每一个瞬间。侵权即删，谢谢支持！

附上汇总贴：小猴编程C++ | 汇总-CSDN博客

---

【题目描述】

等差数列 指从第二项起，每一项与它的前一项的差等于同一个常数的一种数列，且这个常数叫做等差数列的公差。

例如，数列 2,5,8,11 就是一个公差为 3 的等差数列。

先给定一个长度为 n 的数列 a1,a2,…,an，请计算该数列中有多少个长度不小于 3 子段是等差数列。

【输入】

第一行，包含一个整数 n。

第二行，包含 n 个整数 a1,a2,…,an。

【输出】

一行，一个整数，表示结果。

【输入样例】

10
-1 1 3 3 3 2 3 2 1 0

【输出样例】

5

【代码详解】

#include <bits/stdc++.h>
using namespace std;
int n, last2, last1, cnt;
long long sum=0;
int a[100005];
int main()
{
    cin >> n;
    for (int i=1; i<=n; i++) cin >> a[i];
    int d = a[2]-a[1];
    cnt = 2;
    for (int i=3; i<=n; i++) {
        if (a[i]-a[i-1] == d) cnt++;
        else {
            sum += 1ll * (cnt-2)*(cnt-2+1)/2;
            d = a[i]-a[i-1];
            cnt = 2;
        }
    }
    if (cnt>=3) sum += (cnt-2)*(cnt-2+1)/2;
    cout << sum;
    return 0;
}

【运行结果】

10
-1 1 3 3 3 2 3 2 1 0
5