hdu(5400)——Arithmetic Sequence（想法题）

题意：

首先，现在给出三个数字n，d1，d2。然后第二行给出n个数字。

然后题目要求的是求这个序列中有几个区间满足一下条件之一：

1）这个区间是一个等差数列，且公差为d1或d2；

2）若一个区间为[l,r]，那么有l<=i<=r，使得[l,i]范围内是公差为d1的等差数列，[i,r]区间内是公差为d2的等差数列。

*注意，这里单个数字一定是满足等差数列的。而且这里数字最好都使用__int64来保存，因为这个原因我们队WA了好几次。

思路：

这里我设了头和尾两个指针，分别用l和r表示。

我用des保存了相邻两个数组的差值。

当pos==1时，

1）如果des==d1，那么只需要改变r指针就好，然后使计数器ans+=r-l+1；

2）如果des==d2，那么使pos=2，并且同时改变r指针，ans+=r-l+1；

3）如果des!=d1&&des!=d2，那么使左右指针同时指向当前位置i，ans+=r-l+1（因为要加上它的本身）；

当pos==2时，

1）如果des==d2，那么同上1

2）des==d1，因为题目说明了d1是不能出现在d2后面的，所以这又算是另外一个区间了，所以使头指针指向l=i-1，r=i；ans+=r-l+1；同时pos=1；

3）如果des!=d1&&des!=d2，那么使左指针改变为i就好，因为此时满足条件的就只有它本身这一个区间了，所以ans++就好。

最后的答案就是ans

#include<stdio.h>#include<string.h>#include<iostream>#include<algorithm>#include<vector>using namespace std;#define maxn 100010typedef __int64 ll;ll a[maxn],des[maxn];int main(){    int n,d1,d2;    while(~scanf("%d%d%d",&n,&d1,&d2)){        memset(a,0,sizeof(a));        memset(des,0,sizeof(des));        scanf("%I64d",&a[1]);        for(int i=2;i<=n;i++){            scanf("%I64d",&a[i]);            des[i]=a[i]-a[i-1];//des数组中保存的是后一个数与前一个数的差值         }        int l=1,r=0;        int pos=1;        //以下如果pos==1，那么说明现在是公差为d1的//如果pos==2，那么说明现在是公差为d2的         __int64 ans=1;        for(int i=2;i<=n;i++){//从第二个开始判断             if(pos==1){                if(des[i]==d1){                     r=i;                    ans+=r-l+1;                }                else if(des[i]==d2){                    pos=2; r=i;                    ans+=r-l+1;                }                else{                    l=i; r=i;                    ans+=r-l+1;                }            }            else if(pos==2){                if(des[i]==d2){                    r=i;                     ans+=r-l+1;                }                else{                    if(des[i]==d1){                        pos=1;                        l=i-1;//这里注意，因为我们des保存的是a[i]-a[i-1]，所以这里l应该指向i-1这个位置                         r=i;                        ans+=r-l+1;                    }                    else l=i,ans++;                }            }        }        printf("%I64d\n",ans);    }}