little w and Discretization（区间mex 主席树+二分）

登录—专业IT笔试面试备考平台_牛客网

一个长度为n的数组,m次查询，问每次只离散化l~r后，多少数跟以前不一样了

题解：

主席树+二分

按顺序给每个a[i]上（只添加a[1~i]时）建棵线段树 分别维护 数值 1 ~ n
最后一次出现的位置，和每个数出现的次数
查询时二分查询第r棵树

#include<bits/stdc++.h>
#define lowbit(x) ((x)&(-x))
#define int long long
using u64 = unsigned long long;
using namespace std;
std::mt19937_64 rng;
const int N=3e5+10,maxn=1e9+7;
typedef pair<int,int>PII;
vector<int > vs;
int m,n,root[N],cnt=0,a[N];

struct pp
{
	int l,r,mi,sum;
} tr[N*30];

void insert(int &q,int p,int l,int r,int val,int last)
{

	q=++cnt;
	tr[q]=tr[p];
	if(l==r)
	{
		tr[q].mi=last;
		tr[q].sum++;
		return;
	}
	int mid=(l+r)/2;
	if(val<=mid) insert(tr[q].l,tr[p].l,l,mid,val,last);
	else insert(tr[q].r,tr[p].r,mid+1,r,val,last);
	tr[q].sum=tr[tr[q].l].sum + tr[tr[q].r].sum;
	tr[q].mi =min(tr[tr[q].l].mi ,tr[tr[q].r].mi);
}

int query(int q,int p,int l,int r,int lim)
{
	int mid=(l+r)/2;
	if(l==r)
	{
		return tr[q].sum -tr[p].sum;
	}
	if(tr[tr[q].l].mi<lim) return query(tr[q].l,tr[p].l,l,mid,lim);

	else
	{
		/*如果1~mid每个数出现位置的最小值都在l的右边(查询root[r]时r右边的数还没有插入)，那说明1~mid里的所有数都在
		数组位置的l~r 中出现过，那么最小没出现过的数一定在r~mid里
		*/
		return query(tr[q].r,tr[p].r,mid+1,r,lim)+tr[tr[q].l].sum -tr[tr[p].l].sum;
	}
}

signed main()
{
	int i,j,l,r;
	scanf("%lld",&n) ;
	for(i=1; i<=n; i++)
	{
		scanf("%lld",&a[i]);
		insert(root[i],root[i-1],1,n,a[i],i);
	}
	scanf("%lld",&m);
	while(m--)
	{
		scanf("%lld%lld",&l,&r);
		int ans=r-l+1-query(root[r],root[l-1],1,n,l);
		printf("%lld\n",ans);
	}
	return 0;
}