JZOJ 3379. 【NOI2013模拟】查询【主席树】

---

题目：

传送门

---

题意：

问在原序列的基础上，有多个连续的数的集合向合并且有重复性，现在询问得到的新序列的第$p_i$小数

---

分析：

我们通过主席树来实现对新序列的$O(logn)$查询
因为数之间的差距太大，所以我们用离散化来使得程序可行

---

代码：

#pragma GCC optimize("Ofast")
#pragma GCC optimize("inline")
#include<cstdio>
#include<string>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
#define LL long long 
#define LZX ILU
using namespace std;
inline LL read() {
    LL d=0,f=1;char s=getchar();
    while(s<'0'||s>'9'){if(s=='-')f=-1;s=getchar();}
    while(s>='0'&&s<='9'){d=d*10+s-'0';s=getchar();}
    return d*f;
}  
using namespace std;
const int N=200010;
int n,Q,m,k,tot,a[200005],b[200005],root[200005],L[6],R[6];
struct Tree{
	int lc,rc,cnt;
}tree[4400005];
int build(int l,int r)
{
	int p=++tot;
	if(l==r) return p;
	int mid=(l+r)>>1;
	tree[p].lc=build(l,mid);
	tree[p].rc=build(mid+1,r);
	return p;
}
int get(int now,int l,int r,int x)
{
	int p=++tot;
	tree[p]=tree[now];
	tree[p].cnt++;
	if(l==r) return p;
	int mid=(l+r)>>1;
	if(x<=mid) tree[p].lc=get(tree[now].lc,l,mid,x);
	else tree[p].rc=get(tree[now].rc,mid+1,r,x);
	return p;
}
int ask(int nowl[],int nowr[],int l,int r,int k)
{
	if(l==r) return l;
	int mid=(l+r)>>1,s=0;
	for(int i=1;i<=m;i++) s+=tree[tree[nowr[i]].lc].cnt-tree[tree[nowl[i]].lc].cnt;
	if(s>=k)
	{
		for(int i=1;i<=m;i++)
		{
			nowl[i]=tree[nowl[i]].lc;
			nowr[i]=tree[nowr[i]].lc;
		}
		return ask(nowl,nowr,l,mid,k);
	} 
	else
	{
		for(int i=1;i<=m;i++)
		{
			nowl[i]=tree[nowl[i]].rc;
			nowr[i]=tree[nowr[i]].rc;
		}
		return ask(nowl,nowr,mid+1,r,k-s);
	}
}
int main()
{
	n=read();Q=read();
	root[0]=build(1,n);
	for(int i=1;i<=n;i++) a[i]=b[i]=read();
	sort(b+1,b+1+n);
	int T=unique(b+1,b+1+n)-b-1;
	for(int i=1;i<=n;i++)
	{
		a[i]=lower_bound(b+1,b+1+T,a[i])-b;
		root[i]=get(root[i-1],1,n,a[i]);
	}
	while(Q--)
	{
		m=read();k=read();
		for(int i=1;i<=m;i++)
		{
			L[i]=read();L[i]=root[L[i]-1];
			R[i]=read();R[i]=root[R[i]];
		}
		printf("%d\n",b[ask(L,R,1,n,k)]);
	}
	return 0;
}