JZOJ6911. 【2020.12.01提高组模拟】莫队(team)-CSDN博客

本文链接：https://blog.csdn.net/qq_43649416/article/details/110452771

Description

支持单点修改，求区间 $[l, r]$ 有多少子区间不包括重复数字。
$n,q\le2e5$

Solution

裸题？对于每一个节点维护前面第一个与它相同的位置，假设是 $pre_i$ ，那么询问 $[l, r]$ 相当于是求：
$\sum_{i=l}^ri-max_{j\le l}(pre_j,l-1)$
考虑这个东西其实就是把 $l - 1$ 带进 $[l, r]$ 中，求一个前缀 $m a x$ 的单调栈，每个位置的最大值之和。
因此我们有一种用线段树维护最大值的方法。
考虑将 $v$ 带进 $[l, r]$ 求前缀最大值的时候，如果 $v\le max(l,mid)$ ，那么走出 $[l, m i d]$ 之后的最大值就是 $m a x (l, m i d)$ ，因此我们可以对于线段树上每一个节点维护 $m a x (l, m i d)$ 带入 $[m i d + 1, r]$ 的贡献，然后往左区间走。
如果 $v > m a x (l, m i d)$ ，那么 $[l, m i d]$ 内的点都被跳过了，贡献直接求出，然后往右区间走。
这样求解一个 $v$ 在区间 $[l, r]$ 的贡献是 $log^2$ 的。
修改维护同样也是 $log^2$ 的。

还有一种稍微复杂一点的 $O(n\log n)$ 方法。
考虑离线，按照时间建一个线段树，前往后枚举序列位置 $i$ ，维护不同时间的 $max_{j\le i}(pre_j)$ ，考虑当前位置在当前时间分成了若干段，通过打区间max即可完成总共 $O (q)$ 次修改。
对于每个询问的 $l$ 取max也是 $O (1)$ 的修改。
由于求和，还要维护历史和，用吉司机线段树。
不太清楚细节

#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<set>
#define maxn 200005
#define ll long long 
using namespace std;

int n,m,i,j,k,a[maxn],pre[maxn];
set<int> S[maxn];
set<int> ::iterator it,it1;
ll ans;

void read(int &x){
	x=0; char ch=getchar();
	for(;ch<'0'||ch>'9';ch=getchar());
	for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
}

int mx[maxn*4]; ll t[maxn*4];
ll getit(int x,int l,int r,int v){
	if (l==r) return max(v,mx[x]);
	int mid=(l+r)>>1;
	if (v<=mx[x<<1]) return getit(x<<1,l,mid,v)+t[x];
	else return 1ll*v*(mid-l+1)+getit(x<<1^1,mid+1,r,v);
}

void maketree(int x,int l,int r){
	if (l==r) {mx[x]=pre[l],t[x]=0;return;}
	int mid=(l+r)>>1;
	maketree(x<<1,l,mid),maketree(x<<1^1,mid+1,r);
	mx[x]=max(mx[x<<1],mx[x<<1^1]);
	t[x]=getit(x<<1^1,mid+1,r,mx[x<<1]);
}

void change(int x,int l,int r,int p,int v){
	if (l==r) {mx[x]=v,t[x]=0;return;}
	int mid=(l+r)>>1;
	if (p<=mid) change(x<<1,l,mid,p,v);
	else change(x<<1^1,mid+1,r,p,v);
	mx[x]=max(mx[x<<1],mx[x<<1^1]);
	t[x]=getit(x<<1^1,mid+1,r,mx[x<<1]);
}

int now;
void query(int x,int l,int r,int L,int R){
	if (l>R||r<L) return;
	int mid=(l+r)>>1;
	if (L<=l&&r<=R){
		if (l==r) now=max(now,mx[x]),ans-=now; else {
			if (now<=mx[x<<1]){
				query(x<<1,l,mid,L,R),now=mx[x];
				ans-=t[x];
			} else {
				ans-=1ll*now*(mid-l+1);
				query(x<<1^1,mid+1,r,L,R);
			}
		}
		return;
	}
	query(x<<1,l,mid,L,R);
	query(x<<1^1,mid+1,r,L,R);
}

void prepare(){
	static int bz[maxn];
	for(i=1;i<=n;i++)
		pre[i]=bz[a[i]],bz[a[i]]=i;
	maketree(1,1,n);
}

int main(){
	freopen("ceshi.in","r",stdin);
	freopen("ceshi.out","w",stdout);
//	freopen("team.in","r",stdin);
//	freopen("team.out","w",stdout);
	read(n),read(m);
	for(i=1;i<=n;i++) read(a[i]),S[a[i]].insert(i);
	prepare();
	while (m--){
		int opt,x,y; read(opt),read(x),read(y);
		if (opt==1){
			S[a[x]].erase(x);
			it=S[a[x]].upper_bound(x);
			if (it!=S[a[x]].end()) {
				if (it==S[a[x]].begin())
					change(1,1,n,*it,0),pre[*it]=0;
				else {
					it1=it,it1--;
					change(1,1,n,*it,*it1),pre[*it]=*it1;
				}
			}
			it=S[y].upper_bound(x);
			if (it!=S[y].end())
				change(1,1,n,*it,x),pre[*it]=x;
			if (it!=S[y].begin())
				it--,change(1,1,n,x,*it),pre[x]=*it;
			else change(1,1,n,x,0),pre[x]=0;
			a[x]=y,S[y].insert(x);
		} else {
			ans=1ll*x*(y-x+1)+1ll*(y-x+1)*(y-x)/2;
			now=x-1,query(1,1,n,x,y);
			printf("%lld\n",ans);
		}
	}
}