hdu 5400

      题意:先给出三个正整数n,d1,d2,再给出含n个数的序列,求这个序列中有多少个区间满足以下几个条件之一:

                1:这个区间是一个等差数列,且公差为d1或d2;

                2:若区间的下标范围为[l,r],应有l<=i<=r,使得[l,i]范围是公差为d1的等差数列,[i,r]范围是公差为d2的等差数列。

               注意:单个数字是一定符合题意的,因此答案至少为n;

     其实是很水的题。。。。可惜求区间数目求了好久,哎╮(╯▽╰)╭

    代码如下:

#include <cstdio>
#include <stack>
#include <set>
#include <iostream>
#include <string>
#include <vector>
#include <queue>
#include <list>
#include <functional>
#include <cstring>
#include <algorithm>
#include <cctype>
#include <string>
#include <map>
#include <iomanip>
#include <cmath>
#define LL long long
#define ULL unsigned long long
#define SZ(x) (int)x.size()
#define Lowbit(x) ((x) & (-x))
#define MP(a, b) make_pair(a, b)
#define MS(arr, num) memset(arr, num, sizeof(arr))
#define PB push_back
#define F first
#define S second
#define ROP freopen("input.txt", "r", stdin);
#define MID(a, b) (a + ((b - a) >> 1))
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define lrt rt << 1
#define rrt rt << 1|1
#define root 1,n,1
#define BitCount(x) __builtin_popcount(x)
#define BitCountll(x) __builtin_popcountll(x)
#define LeftPos(x) 32 - __builtin_clz(x) - 1
#define LeftPosll(x) 64 - __builtin_clzll(x) - 1
const double PI = acos(-1.0);
const LL INF = (((LL)1)<<62)+1;
using namespace std;
const double eps = 1e-5;
const int MAXN = 300 + 10;
const int MOD = 1000007;
const double M=1e-8;
const int N=100005;
typedef pair<int, int> pii;
typedef pair<int, string> pis;
int n,m,a[N],d1,d2;
LL q[N] , d[N];
inline LL get(LL x)
{
    x+=1;
    return x*(x-1)>>1;
}
int main()
{
    int i,j;
    while(~scanf("%d%d%d",&n,&d1,&d2))
    {
        int t;
        LL ans=n;
        scanf("%d",a);
        for (i=1;i<n;i++) {
            scanf("%d",a+i);
            q[i]=a[i]-a[i-1];
        }
        if (n<=1) {
            puts("1");
            continue;
        }
        int k=0;
        LL cnt1=0,cnt2=0;
        for (i=1;i<n;i++) {
            if (q[i]==d1) {
                if (cnt2) {
                    ans+=get(cnt1)+get(cnt2)+cnt1*cnt2;
                    cnt1=cnt2=0;
                }
                cnt1++;
            }
            else if (q[i]==d2) cnt2++;
            else {
                ans+=get(cnt1)+get(cnt2)+cnt1*cnt2;
                cnt1=cnt2=0;
            }
        }
        ans+=get(cnt1)+get(cnt2)+cnt1*cnt2;
        printf("%lld\n",ans);
    }
}
























评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值