2015 多校联赛 ——HDU5400（水）

Sample Input

5 2 -2
0 2 0 -2 0
5 2 3
2 3 3 3 3

Sample Output

12
5

求最多多少序列满足，前半部分满足d（j+1） = d（j）+d1，后半部分d（j+1）= d（j）+d2，或者只满足其中一个。

假设可以找出长度为x的序列，则有x-1 + x-2 + .... + 1种。所以从头到尾找一遍即可。

#include <iostream>
#include <cstdio>

using namespace std;
typedef long long ll;

int num[101000];
int sta[101000];

ll p(ll n)
{
return (1+n)*n/2;
}

int main()
{
int n, d1, d2;

while(scanf("%d%d%d", &n, &d1, &d2)!=EOF)
{
scanf("%d", num+1);
for(int i=2; i<=n; i++)
{
scanf("%d", num+i);
if(num[i]==(num[i-1]+d1)) sta[i] = 1;
else if(num[i]==(num[i-1]+d2)) sta[i] = 2;
else sta[i] = 0;
}
//        for(int i=1; i<=n; i++) printf("%d ", num[i]);
//        for(int i=1; i<=n; i++) printf("%d ", sta[i]);

ll re = 0;
int cur = 1;
while(cur<=n)
{
int ccur = cur+1;
bool flag = true;
while(ccur<=n)
{
if(sta[ccur]==0) break;
else if(sta[ccur]==1) ccur++;
else if(sta[ccur]==2)
{
while(ccur<=n)
{
if(sta[ccur]==2) ccur++;
else
{
flag = false;
break;
}
}
}
if(!flag) break;
}
re += p(ccur-cur-1);
if((ccur-cur)>1)
cur = ccur - 1;
else cur = ccur;
if(ccur>n) break;
}

printf("%I64d\n", re+n);
}
return 0;
}