洛谷 P3834 【模板】可持久化线段树 2（主席树）

整体二分做法：

所谓整体二分，就是把一整块的询问放在一起进行二分答案。

首先对于数组中原来的数和询问，把它们封装在一个结构体中，这个结构体包括x,y,k,id,opt等。（可能对于更复杂的题还需存储一些其余的值）

对于数组中原来的数，我们发现y,k是没有作用的，x代表的是这个位置的数的大小，id表示数的位置，opt是它的类型。

对于询问，x，y，k表示在区间x~y中查询第k小的数，id表示的是询问的编号，opt是它的类型。

那么接下来我们要对这些封装好的什么东西一起进行二分答案，我们把它们存在数组q里，在接下来的描述中我们管它叫队列。

在solve函数中，ql，qr分别表示当前处理的队列的子队列的左端点和右端点。l，r是我们二分的权值。首先判断l是不是与r相等，若是相等说明找到了，此时对于目前处理的队列中所有询问而言结果都是l（即r）。

如果没有满足相等条件，那么我们还需要继续二分下去。mid=(l+r)/2，我们把当前维护的队列中所有opt=1的类型中x值小于等于mid的划分到左边，同时累加个数（这里推荐用小常数的树状数组），而如果大于mid就划分到右边。

对于opt=2的元素而言，我们用树状数组统计x，y之间的比mid小的数的个数，然后将个数与k进行比较。如果个数比k小，那么把它划分到同样把它划分早左边，反之划分到右边。每次结束时，我们需要把之前累加的树状数组还原。

需要注意的是，l，r，mid是我们二分出来的答案的数值，ql，qr是当前子队列的左右断点。

#include <bits/stdc++.h>
#define lowbit(x) x&(-x)
using namespace std;
const int N=2e5+5;
int n,m,x,y,k,minn,maxn,cnt;
int c[N],ans[N];
struct node{int x,y,k,id,opt;}q[N<<1],q1[N<<1],q2[N<<1];

inline void change(int x,int v)
{
	while (x<=n)
	{
		c[x]+=v;
		x+=lowbit(x);
	}
}
inline int query(int x)
{
	int res=0;
	while (x)
	{
		res+=c[x];
		x-=lowbit(x);
	}
	return res;
}

void solve(int ql,int qr,int l,int r)
{
	if (ql>qr) return;
	if (l==r)
	{
		for (register int i=ql; i<=qr; ++i) if (q[i].opt==2) ans[q[i].id]=l;
		return;	
	}
	int mid=l+r>>1;
	int p1=0,p2=0;
	for (register int i=ql; i<=qr; ++i)
	if (q[i].opt==1)
	{
		if (q[i].x<=mid)
		{
			change(q[i].id,1);
			q1[++p1]=q[i];	
		}
		else q2[++p2]=q[i];
	}
	else
	{
		int res=query(q[i].y)-query(q[i].x-1);
		if (q[i].k<=res) q1[++p1]=q[i];
		else 
		{
			q[i].k-=res;
			q2[++p2]=q[i];
		}
	}
	for (register int i=1; i<=p1; ++i) if (q1[i].opt==1) change(q1[i].id,-1);
	for (register int i=1; i<=p1; ++i) q[ql+i-1]=q1[i];
	for (register int i=1; i<=p2; ++i) q[ql+p1-1+i]=q2[i];
	solve(ql,ql+p1-1,l,mid);
	solve(ql+p1,qr,mid+1,r);
}

int main(){
	scanf("%d%d",&n,&m);
	minn=1e9; maxn=-1e9;
	for (register int i=1; i<=n; ++i)
	{
		scanf("%d",&x);
		minn=min(minn,x);
		maxn=max(maxn,x);
		q[++cnt]=(node){x,0,0,i,1};
	}
	for (register int i=1; i<=m; ++i)
	{
		scanf("%d%d%d",&x,&y,&k);
		q[++cnt]=(node){x,y,k,i,2};
	}
	solve(1,cnt,minn,maxn);
	for (register int i=1; i<=m; ++i) printf("%d\n",ans[i]);
return 0;	
}