hdu 5400 Arithmetic Sequence（计数）

最新推荐文章于 2021-03-31 18:09:34 发布

原创最新推荐文章于 2021-03-31 18:09:34 发布 · 690 阅读

0 ·

CC 4.0 BY-SA版权

GRADE:D 同时被 3 个专栏收录

1021 篇文章

订阅专栏

HDU

582 篇文章

订阅专栏

数学-计数问题

67 篇文章

订阅专栏

本文介绍了解决HDU5400 Arithmetic Sequence问题的算法，通过双指针技巧和数学分析，实现了O(N)的时间复杂度。详细解释了如何在数组中寻找符合条件的连续子序列，并计算这些子序列的数量。

题目链接：hdu 5400 Arithmetic Sequence

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;
const int maxn = 1e5 + 5;
typedef long long ll;
int N, D1, D2, A[maxn];

int main () {
	while (scanf("%d%d%d", &N, &D1, &D2) == 3) {
		for (int i = 1; i <= N; i++) scanf("%d", &A[i]);
		ll ans = N;
		int l = 1, r = 1;
		while (l <= N) {
			while (r < N && A[r] + D1 == A[r+1]) r++;
			while (r < N && A[r] + D2 == A[r+1]) r++;
			if (r == l) {
				r++, l++;
				continue;
			}
			int n = r - l;
			ans = ans + 1LL * n * (n+1) / 2;
			l = r;
		}
		printf("%lld\n", ans);
	}
	return 0;
}