码迷,mamicode.com
首页 > 其他好文 > 详细

hdu5400Arithmetic Sequence

时间:2017-06-13 20:41:06      阅读:144      评论:0      收藏:0      [点我收藏+]

标签:content   typedef   include   return   code   lld   pre   art   情况   

//一个序列,两个公差d1,d2
//问有多少个区间使得这个区间存在一个点,它的左边是公差为d1的序列
//它的右边是公差为d2的序列
//直接存入每一个点向左和向右延伸的公差长度,乘一下即可
//还有就是注意一下d1=d2的情况
#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std ;
const int maxn = 1e5+10 ;
int a[maxn] ;
typedef long long ll ;
ll l[maxn] , r[maxn] ;
int main()
{
    int n , d1 , d2 ;
    while(~scanf("%d%d%d" ,&n , &d1 , &d2))
    {
        for(int i = 1;i <= n;i++)
            scanf("%d" , &a[i]) ;
        ll ans = 0 ;
        if(d1 == d2)
        {
            ll sum = 1;
            for(int i = 2;i <= n;i++)
            if(a[i] == a[i-1] + d1)
            sum++ ;
            else
            {
                ans += (sum+1)*sum/2 ;
                sum = 1 ;
            }
            ans += (sum+1)*sum/2 ;
        }
        else
        {
            l[0] = 0 ;r[n+1] = 0 ;
            for(int i = 1;i <= n;i++)
            if(a[i] == a[i-1] + d1)
            l[i] = l[i-1] + 1 ;
            else l[i] = 1 ;
            for(int i = n;i >= 1;i--)
            if(a[i] == a[i+1] - d2)
            r[i] = r[i+1] + 1 ;
            else r[i] = 1 ;
            for(int i = 1;i <= n;i++)
            ans += l[i]*r[i] ;
        }
        printf("%lld\n" , ans) ;
    }
    return 0 ;
}

hdu5400Arithmetic Sequence

标签:content   typedef   include   return   code   lld   pre   art   情况   

原文地址:http://www.cnblogs.com/yutingliuyl/p/7003120.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!