Arithmetic Sequence(多校第九场)

原题链接:http://acm.hdu.edu.cn/showproblem.php?pid=5400

这题其实就是求有多少公差为d1和公差为d2的等差序列和前半段是公差为d1后半段的公差为d2的序列。。。。。

我们可以用个b数组保存序列中相邻两项的和,然后直接查找b数组的值是否等于d1或d2。再来进行处理。。

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm> 
#define maxn 100000
int a[maxn],b[maxn];
using namespace std;
int main()
{
    int n,d1,d2;
    while(scanf("%d %d %d",&n,&d1,&d2)!=EOF)
    {
        scanf("%d",&a[0]);
        for(int i=1;i<n;i++)
        {
            scanf("%d",&a[i]);
            b[i-1]=a[i]-a[i-1];
        }
        __int64 s1=0,s2=0,sum=0;
        for(int i=0;i<n-1;i++)
        {
            if(b[i]==d1)
            {
            if(b[i-1]!=d1)//b[i-1]不为d1,那么s1=0
             s1=0;
            s1++;
            sum+=s1;//sum加上当前公差为的d1序列长度
            s2=0;//s2进行清零
            }
            else if(b[i]==d2)
            {
            s2++;
            sum+=s2+s1;//加上公差为d2和前半段为d1后半段为d2的序列长度
            }
            else
            s1=s2=0;
        }
        printf("%I64d\n",sum+n);
    }
    return 0;
} 

 

posted @ 2015-08-18 22:26  __NaCl  阅读(154)  评论(0编辑  收藏  举报