看着qscdalao的视频,初步学习了主席树
自己手动模拟了一遍,才理解线段树是怎么建的
代码如下
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=1e5+7;
int n,m,x,y,k,root[maxn],a[maxn],ha[maxn],hs,cnt;
struct Node
{
int l,r,sum;
}T[maxn*40];
int getid(int x)
{
return lower_bound(ha,ha+hs,x)-ha+1;
}
void Update(int l,int r,int &x,int y,int pos)
{
T[++cnt]=T[y]; T[cnt].sum++; x=cnt;
if(l==r) {
return ;
}
int mid=l+r>>1;
if(pos<=mid) {
Update(l,mid,T[x].l,T[y].l,pos);
}
else{
Update(mid+1,r,T[x].r,T[y].r,pos);
}
}
int Query(int l,int r,int x,int y,int k)
{
if(l==r) {
return l;
}
int mid=l+r>>1;
int sum=T[T[y].l].sum-T[T[x].l].sum;
if(sum>=k) {
Query(l,mid,T[x].l,T[y].l,k);
}
else{
Query(mid+1,r,T[x].r,T[y].r,k-sum);
}
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) {
scanf("%d",&a[i]);
ha[i-1]=a[i];
}
sort(ha,ha+n);
hs=unique(ha,ha+n)-ha;
for(int i=1;i<=n;i++) {
Update(1,n,root[i],root[i-1],getid(a[i]));
}
for(int i=1;i<=m;i++) {
scanf("%d%d%d",&x,&y,&k);
printf("%d\n",ha[Query(1,n,root[x-1],root[y],k)-1]);
}
}