Arithmetic Sequence(多校第九场)

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm> 
#define maxn 100000
int a[maxn],b[maxn];
using namespace std;
int main()
{
    int n,d1,d2;
    while(scanf("%d %d %d",&n,&d1,&d2)!=EOF)
    {
        scanf("%d",&a[0]);
        for(int i=1;i<n;i++)
        {
            scanf("%d",&a[i]);
            b[i-1]=a[i]-a[i-1];
        }
        __int64 s1=0,s2=0,sum=0;
        for(int i=0;i<n-1;i++)
        {
            if(b[i]==d1)
            {
            if(b[i-1]!=d1)//b[i-1]不为d1，那么s1=0
             s1=0;
            s1++;
            sum+=s1;//sum加上当前公差为的d1序列长度
            s2=0;//s2进行清零
            }
            else if(b[i]==d2)
            {
            s2++;
            sum+=s2+s1;//加上公差为d2和前半段为d1后半段为d2的序列长度
            }
            else
            s1=s2=0;
        }
        printf("%I64d\n",sum+n);
    }
    return 0;
}