您的位置:首页 > 其它

hdu 5400 等差数列

2015-08-18 18:27 176 查看
#include<stdio.h>
#include<string>
#include<map>
#include<vector>
#include<cmath>
#include<stdlib.h>
#include<string.h>
#include<algorithm>
#include<queue>
#include<iostream>
using namespace std;
const int N=1e5+10;
const int MOD=1e9+7;
const int inf=1e9+7;
int n,m;
int a
;
int d1,d2;
int b
,c
;
int main(){
#ifndef ONLINE_JUDGE
freopen("aaa","r",stdin);
#endif
int T;
while(~scanf("%d%d%d",&n,&d1,&d2)){
for(int i=1;i<=n;i++)scanf("%d",&a[i]);
b[1]=1;
for(int i=2;i<=n;i++)  if(a[i]-a[i-1]==d1) b[i]=b[i-1]+1;else b[i]=1;
c
=1;
for(int i=n-1;i>=1;i--) if(a[i+1]-a[i]==d2) c[i]=c[i+1]+1;else c[i]=1;
if(d1==d2) for(int i=1;i<=n;i++) b[i]=1;
long long res=0;
for(int i=1;i<=n;i++) res+=(long long)b[i]*c[i];
printf("%I64d\n",res);
}
return 0;
}


记得考虑long long 和 d1=d2的情况,其他没有什么
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: