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 100010
typedef __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);
    }
}