hdu5400

来源：互联网发布：出租房软件编辑：程序博客网时间：2024/05/18 06:45

根据题意，用数组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;}

0 0