HDU 5400 Arithmetic Sequence

題目鏈接

題意：

n個數，兩個公差d1, d2

求符合要求的子序列 [l, r](1≤l≤r≤n) 數量

要求：存在點i（l <= i <= r）使得a[l] ~a[i]是公差爲d1的等差數列，a[i]~a[r]是公差爲d2的等差數列

思路：枚舉每個點，算出每個點向左向右符合條件的數有幾個（自己本身算一個）記爲 l[i] 和 r[i] 

l[i] * r[i] 就是以i點爲分割點符合條件的子序列個數

代碼如下：

#include<cstdio>
#include<cstring>
#include<string>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
const int N = 1e5+10;
int n, d1, d2;
ll a[N], l[N], r[N];
ll ans, cnt;

int main()
{
	int i, j, k;
	while(~scanf("%d%d%d", &n, &d1, &d2))
	{
		for(i = 1; i <= n; i++)
		{
			scanf("%I64d", &a[i]);
		}
		if(d1 == d2)
		{
			for(int i = 1; i <= n; i++) l[i] = 1;
		} 
		else
		{
			l[1] = 1;
			for(int i = 2; i <= n; i++)
			{
				if((a[i] - a[i-1]) == d1)
				{
					l[i] = l[i-1] + 1;
				}
				else l[i] = 1;
			}
		}
		ans = 0;
		r[n] = 1;
		ans += l[n] * r[n];
		for(int i = n - 1; i > 0; i--)
		{
			if((a[i+1] - a[i]) == d2)
			{
				r[i] = r[i+1] + 1;
			}
			else r[i] = 1;
			ans += l[i] * r[i];
		}
		printf("%I64d\n", ans);
	}
	return 0;
}

i(1≤i≤n)i(1≤i≤n)在