little w and Discretization--可持久化线段树

题目链接:

https://ac.nowcoder.com/acm/contest/20647/C

题意：

给定一个长度为n的数组，m次询问，每次询问把一个区间的数字进行离散化，问离散化之后与原本数字不相同的数字个数有多少个。每个询问之间是相互独立的。如对{100,200,500,200,300}进行离散化后将会变成{1,2,4,2,3}，离散化前后不相同的数字有5个。

样例:

输入

9
4 3 1 2 5 3 3 1 4
8
1 3
2 4
2 5
1 5
3 7
4 7
4 8
5 9

输出

2
0
1
0
1
4
1
4

样例解释:
第三个查询表示的子数组为{3,1,2,5}，将其离散化后得到{3,1,2,4}，有一个元素元素发生变化。
第四个查询表示的子数组为{4,3,1,2,5}，将其离散化后得到{4,3,1,2,5}，没有任何元素元素发生变化。
第五个查询表示的子数组为{1,2,5,3,3}，将其离散化后得到{1,2,4,3,3}，有一个元素元素发生变化。
第六个查询表示的子数组为{2,5,3,3}，将其离散化后得到{1,3,2,2}，有四个元素元素发生变化

思路：

首先，对一个区间进行离散化，会变得数字是什么，是大于区间MEX的数，MEX是区间最小没有出现的数，如1，2，3，5，6的MEX就是4。离散化之后5和6都会与原来不同，所以问题便转化成了求出区间[l,r]内的MEX,再根据MEX求出[l,r]区间内在1~MEX之间的数，这些数离散化之后是不变的，变化的数就是 (r-l+1)-不变的数。

求出[l,r]区间内在1~MEX之间的数这个不必多说了，直接套主席树的板子就可以。这里说一下如何求区间[l,r]的MEX，对于固定的r，我们用数组mi保存每个数最后一次出现的位置，初始值为0，表示没出现过，root[r]的对应的一系列mi数组中就保存了数组1-r中所有数最后一次出现的位置。

查询区间[l,r]的MEX时，查询root[r]版本下最后一次出现位置小于l的最小的数。

代码：

#include <bits/stdc++.h>

using namespace std;

const int N=3e5+10;
const int M=20*N;
int n,m;
int lch[M],rch[M],mi[M],sum[M];
int root[N],a[N],tot=0;

void cpy(int from,int to){
	lch[to]=lch[from];
	rch[to]=rch[from];
	sum[to]=sum[from]; 
}
void insert(int &u,int old,int L,int R,int x,int pos){
	u=++tot;
	cpy(old,u);
	sum[u]++;
	if(L==R){
		mi[u]=pos;
		return ;
	}
	int mid=(L+R)>>1;
	if(x<=mid)
		insert(lch[u],lch[u],L,mid,x,pos);
	else
		insert(rch[u],rch[u],mid+1,R,x,pos);
	mi[u]=min(mi[lch[u]],mi[rch[u]]);
}
int queryMEX(int u,int L,int R,int l){
	if(L==R)
		return L;
	int mid=(L+R)>>1;
	if(mi[lch[u]]<l)
		return queryMEX(lch[u],L,mid,l);
	else
		return queryMEX(rch[u],mid+1,R,l);
}
int queryVal(int s,int t,int L,int R,int l,int r){
	if(L==l&&R==r){
		return sum[t]-sum[s]; 
	}
	int mid=(L+R)>>1;
	if(r<=mid){
		return queryVal(lch[s],lch[t],L,mid,l,r);
	}
	else if(l<=mid){
		int res=0;
		res+=queryVal(lch[s],lch[t],L,mid,l,mid);
		res+=queryVal(rch[s],rch[t],mid+1,R,mid+1,r);
		return res;
	}
	else
		return queryVal(rch[s],rch[t],mid+1,R,l,r);
}
int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;i++){
		scanf("%d",&a[i]);
		if(a[i]>n)
			a[i]=n+1;
		insert(root[i],root[i-1],1,n+1,a[i],i);
	}
	scanf("%d",&m);
	int l,r,mex,res;
	while(m--){
		scanf("%d%d",&l,&r);
		mex=queryMEX(root[r],1,n+1,l);
		res=queryVal(root[l-1],root[r],1,n+1,1,mex);
		res=r-l+1-res;
		printf("%d\n",res);
	}
}