洛谷3834 可持久化线段树(主席树模板)

传送门

【题目分析】

主席树经典问题,静态查询区间第k大数。

拖了很久都没有写的模板。。。。结果差不多此次%你赛都考,痛定思痛后,决定好好研究下。

主席树可以近似理解为权值线段树的前缀和的形式,详细讲解可参考wcr写的主席树讲解:传送门

回到该题,因为不涉及修改操作,所以我们建立n棵权值线段树,先将数据离散化,这样主席树的权值就在1~len中了,每次插入操作就相当于将a[i]到根的路径上所有节点的sun+1,由于只会影响一条链,所以一个节点最多只会有一个儿子发生变化,所以建图大概是这样的:

如上图,如果a[i]为4,那么1~4这条路径上所有节点sum+1,发现1号节点的左儿子其实与新节点左儿子一样,所以直接先将1号节点信息赋给新节点,再递归修改右儿子。

查询时,因为所有权值线段树的值域相同,所以可以进行减法这种操作,具体实现参照代码。

【代码~】

#include<bits/stdc++.h>
using namespace std;
const int MAXN=2e5+10;

int n,q,cnt;
int a[MAXN],ls[MAXN],rt[MAXN];
struct Tree{
	int l,r;
	int sum;
}tr[MAXN*50];

int Read()
{
	int i=0,f=1;
	char c;
	for(c=getchar();(c>'9'||c<'0')&&c!='-';c=getchar());
	if(c=='-')
	  f=-1,c=getchar();
	for(;c>='0'&&c<='9';c=getchar())
	  i=(i<<3)+(i<<1)+c-'0';
	return i*f;
}

void insert(int x,int &root,int l,int r,int key)
{
	root=++cnt;
	tr[root]=tr[x];
	tr[root].sum++;
	if(l==r)
	  return ;
	int mid=l+r>>1;
	if(key<=mid)
	  insert(tr[x].l,tr[root].l,l,mid,key);
	else
	  insert(tr[x].r,tr[root].r,mid+1,r,key);
}

int query(int root1,int root2,int l,int r,int k)
{
	if(l==r)
	  return l;
	int mid=l+r>>1,x=tr[tr[root2].l].sum-tr[tr[root1].l].sum;
	if(k<=x)
	  return query(tr[root1].l,tr[root2].l,l,mid,k);
	else
	  return query(tr[root1].r,tr[root2].r,mid+1,r,k-x);
}

int main()
{
	n=Read(),q=Read();
	for(int i=1;i<=n;++i)
	  a[i]=ls[i]=Read();
	sort(ls+1,ls+n+1);
	int len=unique(ls+1,ls+n+1)-ls-1;
	for(int i=1;i<=n;++i)
	  a[i]=lower_bound(ls+1,ls+len+1,a[i])-ls;
	for(int i=1;i<=n;++i)
	  insert(rt[i-1],rt[i],1,len,a[i]);
	while(q--)
	{
		int l=Read(),r=Read(),k=Read();
		printf("%d\n",ls[query(rt[l-1],rt[r],1,len,k)]);
	}
	return 0;
}

 

posted @ 2018-10-29 09:23  Ishtar~  阅读(153)  评论(0编辑  收藏  举报