题意:
首先,现在给出三个数字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);
}
}