洛谷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;
}