查詢區間第K大,而且沒有修改。
主席樹的原理就是在現有的一顆線段樹上不斷加入新的節點,而加入的對於現有線段樹的影響另開一條鏈記錄下來,這樣我們就可以保存線段樹的歷史版本,在查詢時只需要把區間減一下就可以了。
#include <vector>
#include <cstdio>
#include <algorithm>
using namespace std;
const int maxn = 1e5+7;
vector<int>q;
int a[maxn],root[maxn],cnt=0;
struct node{int l,r,sum;}T[maxn*20];
int getid(int x){return lower_bound(q.begin(),q.end(),x)-q.begin()+1;}
void update(int l,int r,int x,int &y,int pos)
{
T[++cnt]=T[x];T[cnt].sum++;y=cnt; //開闢新的鏈
if(l==r) return ;
int m = (l+r)>>1;
if(m>=pos) update(l,m,T[x].l,T[y].l,pos);//修改新開闢的鏈
else update(m+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 m = (l+r)>>1;
int sum = T[T[y].l].sum-T[T[x].l].sum;
if(sum>=k) query(l,m,T[x].l,T[y].l,k);
else query(m+1,r,T[x].r,T[y].r,k-sum);
}
int main()
{
int i,n,m,k,x,y;
scanf("%d%d",&n,&m);
for(i=1;i<=n;i++) scanf("%d",a+i),q.push_back(a[i]);
sort(q.begin(),q.end());q.erase(unique(q.begin(),q.end()),q.end());
for(i=1;i<=n;i++)
update(1,n,root[i-1],root[i],getid(a[i]));
for(i=1;i<=m;i++)
{
scanf("%d%d%d",&x,&y,&k);
printf("%d\n",q[query(1,n,root[x-1],root[y],k)-1]);
}
return 0;
}