LOJ #2275. 「JXOI2017」颜色

题目链接：传送门

求方案数，乍一看挺吓人，以为是什么思维计数题
其实不然，对于序列上的计数
我们有一个通用的办法，就是通过枚举端点来统计答案
按题意，要求删除之后剩下的是一个完整的区间
正好可以枚举端点，所以我们枚举右端点来计算能有多少个左端点满足条件
由于一次删除的是一个颜色，所以记下颜色的左右区间可以帮助我们快速判断很多信息
若当前枚举到的点就是这个点颜色最后一次出现的位置
　那么从它第一次出现的位置$+1$到现在的位置都是不能作为左端点的
　可以自己画图理解一下，因为第一次出现的位置总会被删除，区间也不会连续
　这个不能取可以用线段树维护，直接区间赋0就可以
否则这个颜色在后边还出现过
　我们记在最后一次之前出现的位置为pre
　那么左端点不能取到pre之前，这个pre可以用一个栈来维护
　因为我们的右端点是单调递增的
理解至上，多画图自己试一下就可以

#include 
#define A 300010

using namespace std;
typedef long long ll;
struct node {int l, r, w, f;}tree[A << 2];
int T, n, a[A], first[A], last[A];
void build(int k, int l, int r) {
	tree[k].l = l; tree[k].r = r; tree[k].w = r - l + 1; tree[k].f = 0;
	if (l == r) return;
	int m = (l + r) >> 1;
	build(k << 1, l, m); build(k << 1 | 1, m + 1, r);
	tree[k].w = tree[k << 1].w + tree[k << 1 | 1].w;
}
void down(int k) {
	tree[k << 1].f = tree[k << 1 | 1].f = 1;
	tree[k << 1].w = tree[k << 1 | 1].w = tree[k].f = 0;
}
void change(int k, int l, int r) {
	if (tree[k].l >= l and tree[k].r <= r) {
		tree[k].w = 0; tree[k].f = 1;
		return;
	}
	if (tree[k].f) down(k);
	int m = (tree[k].l + tree[k].r) >> 1;
	if (l <= m) change(k << 1, l, r);
	if (r > m) change(k << 1 | 1, l, r);
	tree[k].w = tree[k << 1].w + tree[k << 1 | 1].w;
}
int ask(int k, int l, int r) {
	if (tree[k].l >= l and tree[k].r <= r) return tree[k].w;
	if (tree[k].f) down(k);
	int m = (tree[k].l + tree[k].r) >> 1, ans = 0;
	if (l <= m) ans += ask(k << 1, l, r);
	if (r > m) ans += ask(k << 1 | 1, l, r);
	return ans;
}
template<class T> void read(T &x) {
	x = 0; char ch = getchar();
	while (!isdigit(ch)) ch = getchar();
	while (isdigit(ch)) x = x * 10 + ch - '0', ch = getchar();
}
pair<int, int> sta[A];

int main(int argc, char const *argv[]) {
	cin >> T;
	while (T--) {
		int top = 0; ll ans = 0; read(n); build(1, 1, n);
		for (int i = 1; i <= n; i++) first[i] = 0, last[i] = 0;
		for (int i = 1; i <= n; i++) {
			read(a[i]);
			if (!first[a[i]]) first[a[i]] = i;
			last[a[i]] = i;
		}
		for (int i = 1; i <= n; i++) {
			if (i == last[a[i]] and first[a[i]] != last[a[i]]) change(1, first[a[i]] + 1, last[a[i]]);
			else sta[++top] = make_pair(a[i], i);
			while (top and last[sta[top].first] <= i) top--;
			int l = top ? sta[top].second : 0;
			if (i != l) ans += ask(1, l + 1, i);
		}
		printf("%lld\n", ans);
	}
	return 0;
}