题意:
首先,现在给出三个数字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); } }
版权声明:本文为博主原创文章,未经博主允许不得转载。
hdu(5400)——Arithmetic Sequence(想法题)
原文地址:http://blog.csdn.net/acmer_hades/article/details/47796117