C++-POJ2104-K-th Number[主席树]

主席树，大家这么叫的，也不知道为什么？

算法思路：

1.先排序，离散化

2.按排名插入n个版本的线段树

3.没有修改，查询即可

由于维护的是权值的区间和，

因此只要查询版本r和l-1，做差比较即可

 

#include <cstdio>
#include <algorithm>
using namespace std;
const int MAXN=100010;
struct Node{int L,R,sum;};
struct Data{int x,id,rank;};
bool cmp(Data A,Data B){return A.x<B.x;} 
int rt[MAXN],cnt,n,m;
Node T[MAXN<<5];
Data a[MAXN];
void insert(int a,int&x,int L,int R) {
    T[++cnt]=T[x],x=cnt,T[x].sum++;
    if(L==R)return;int mid=(L+R)>>1;
    if(a<=mid)insert(a,T[x].L,L,mid);
    else insert(a,T[x].R,mid+1,R);
}
int query(int l,int r,int k,int L,int R){
    if(L==R)return L;int mid=(L+R)>>1;
    int t=T[T[r].L].sum-T[T[l].L].sum;
    if(k<=t)return query(T[l].L,T[r].L,k,L,mid);
    else return query(T[l].R,T[r].R,k-t,mid+1,R);
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)scanf("%d",&a[i].x),a[i].id=i;
    sort(a+1,a+n+1,cmp);for(int i=1;i<=n;i++)a[a[i].id].rank=i;
    for(int i=1;i<=n;i++)rt[i]=rt[i-1],insert(a[i].rank,rt[i],1,n);
    for(int l,r,k; m--;)scanf("%d%d%d",&l,&r,&k),printf("%d\n",a[query(rt[l-1],rt[r],k,1,n)].x);
    return 0;
}