luogu P3834 【模板】可持久化线段树 1（主席树）

背景：

发现主席树原来搞得不是很好。

题目传送门：

https://www.luogu.org/problemnew/show/P3834

题意：

维护一个支持询问区间第$k$大的东西。

思路：

区间第$k$大，显然我们想到了主席树。
然后你建立一棵动态开点的权值线段树用前缀和优化一下即可。
网上的讲解很多 ，我肯定是不会写数据结构的讲解 。
其实就是存一份模板。
时空复杂度：$\Theta (n\log n)$

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#define INF 1e9+1
using namespace std;
	int ls[8000010],rs[8000010],root[200010],tot[8000010],d[200010];
	int n,m,len=0;
void change(int last,int &now,int x,int l,int r)
{
	if(!now) now=++len;
	tot[now]=tot[last]+1;
	if(l==r) return;
	int mid=(l+r)/2;
	if(x<=mid)
	{
		rs[now]=rs[last];
		change(ls[last],ls[now],x,l,mid);
	}
	else
	{
		ls[now]=ls[last];
		change(rs[last],rs[now],x,mid+1,r);
	}
}
int solve(int x,int y,int k,int l,int r)
{
	if(l==r) return l;
	int tmp=tot[ls[y]]-tot[ls[x]],mid=(l+r)/2;
	return k<=tmp?solve(ls[x],ls[y],k,l,mid):solve(rs[x],rs[y],k-tmp,mid+1,r);
}
int main()
{
	int x,y,k;
	scanf("%d %d",&n,&m);
	for(int i=1;i<=n;i++)
		scanf("%d",&d[i]);
	for(int i=1;i<=n;i++)
		change(root[i-1],root[i],d[i],-INF,INF);
	for(int i=1;i<=m;i++)
	{
		scanf("%d %d %d",&x,&y,&k);
		printf("%d\n",solve(root[x-1],root[y],k,-INF,INF));
	}
}