CF1418G

很厉害的题啊，不会做www

有一种很简洁的 Hash 做法捏。

先放宽要求，考虑出现的次数都是 \(3\) 的倍数。

维护 \(cnt_{i,j}\) 表示前 \(i\) 个数，\(j\) 出现次数模 \(3\) 的值。

那么区间 \([l,r]\) 符合要求当且仅当 \(cnt_{l-1}=cnt_r\)，哈希即可。

那么原要求是恰好为 \(3\)，扫描线即可。

具体细节看代码。

Code：

#include <bits/stdc++.h>
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
const int N = 500005;
int n;
int pre[N], lst[N];
int a[N], cnt[N];
ull val[N], H[N];
map <ull, int> mp;

int main() {
	mt19937 rng(time(0));
	scanf("%d", &n);
	for (int i = 1; i <= n; ++i) scanf("%d", &a[i]);
	for (int i = 1; i <= n; ++i) val[i] = rng() * (1ull << 32) + rng();
	for (int i = 1; i <= n; ++i) {
		cnt[a[i]] = (cnt[a[i]] + 1) % 3;
		if (cnt[a[i]]) H[i] = H[i - 1] + val[a[i]];
		else H[i] = H[i - 1] - 2 * val[a[i]];
	}
	for (int i = 1; i <= n; ++i) pre[i] = lst[a[i]], lst[a[i]] = i;
	mp[0] = 1;
	int p = -1; ll ans = 0;
	for (int i = 1; i <= n; ++i) {
		int t = pre[pre[pre[i]]] - 1;
		while (p < t) ++p, --mp[H[p]];
		ans += mp[H[i]], ++mp[H[i]];
	}
	printf("%lld", ans);
	return 0;
}