Codeforces1352E Special Elements (思维 尺取法)

题目链接: Special Elements

大致题意

给定长度为$n$的序列$a$, 问有多少个$a_i$满足$a_i=a[l:r],1\le l<r\le n$ .

解题思路

思维

我们需要计算的是, 某个位置的数字是否等于一个区间和, 我们很容易想到处理出$a[]$的前缀和.

然后我们可以通过枚举$r$, 来判断之前是否出现对应的$l$.

但是题目卡了哈希表, 又卡了空间, 我们无法存储高效的进行判断.

---

尺取

我们考虑由于$a_i\in [1,n]$, 因此对于每个$r$而言, $l$是单调的, 因此我们可以维护双指针$l,r$表示当前答案区间.

AC代码

#include <bits/stdc++.h>
#define rep(i, n) for (int i = 1; i <= (n); ++i)
using namespace std;
typedef long long ll;
const int N = 8E3 + 10;
int n;
int a[N];
bool fact(int pos) {
	int sum = a[1] + a[2];
	for (int l = 0, r = 2; r <= n; sum += a[++r]) {
		while (r - l >= 2 and sum - a[l] >= a[pos]) sum -= a[l++];
		if (sum == a[pos]) return 1;
	}
	return 0;
}
int main()
{
    int t; cin >> t;
	while (t--) {
		scanf("%d", &n);
		rep(i, n) scanf("%d", &a[i]);

		int res = 0;
		rep(i, n) {
			res += fact(i);
		}

		printf("%d\n", res);
	}

	return 0;
}

END