hdu5400

根据题意,用数组ch[i]记录a[i]和a[i - 1]的差值, len[i]记录到a[i]为止形成的最长的序列长度。根据题意len[i] = len[i-1] + 1(即a[i] 可以添加的前面的序列)的情况有:

1. ch[i] 等于 ch[i-1] 且ch[i] 等于d1或d2。 因为根据题意公差为d1或d2的等差序列都满足条件。

        2. ch[i] 等于 d2 而且 ch[i-1] = d1。

如果上面两种情况不满足,那么当ch[i] 等于 d1或d2时, len[i] = 2, 表示这个数可以与前面一个数组成满足条件的序列。

否则, len[i] = 1;

因为有a[i]结尾的满足条件的序列的个数等于len[i], 最后把这个len[i]累加起来就是答案。代码如下:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>

using namespace std;

const int MAXN= 100002;

int a[MAXN], m, n, b1, b2, len[MAXN], ch[MAXN];
long long ans;

int main()
{
    int i;
    while(scanf("%d%d%d", &n, &b1, &b2) != -1)
    {
        for(i = 1; i <= n; i++)
        {
            scanf("%d", &a[i]);
        }
        ans = 0;
        len[1] = 1;
        for(i = 2; i <= n; i++)
        {
            ch[i] = a[i] - a[i-1];
            if(i == 2)
            {
                if(a[2]-a[1] == b1 || a[2] - a[1] == b2)
                    len[i] = 2;
                else len[i] = 1;
            }
            else
            {
                if(ch[i] == ch[i-1])
                {
                    if(ch[i] == b1 || ch[i] == b2)
                        len[i] = len[i-1] + 1;
                    else
                        len[i]  = 1;
                }
                else if(ch[i] == b2 && ch[i-1] == b1)
                {
                    len[i] = len[i-1] + 1;
                }
                else if(ch[i] == b1 || ch[i] == b2)
                {
                    len[i]  = 2;
                }
                else
                {
                    len[i] = 1;
                }
            }
        }
        for(i = 1; i<= n; i++)
        {
            ans += len[i];
        }
        printf("%I64d\n", ans);
    }
    return 0;
}



评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值