poj2104 K-th Number（主席树模板）

首先离散化，然后对于每个前缀1…i,建立一颗权值线段树。那么l…r的区间信息，就是tree[r]-tree[l-1]。然后因为直接开肯定开不下，我们发现第i棵线段树和第i-1棵线段树基本相同，只有一条链被更改了，因此我们可以利用这一点减少空间花销。也就是说对于每一棵线段树，我只新建出这次被修改了的一条链，所以总的空间复杂度是O(nlogn)的。时间复杂度还是O(nlogn)的。

#include <cstdio>#include <cstring>#include <algorithm>using namespace std;#define ll long long#define N 100010#define inf 0x3f3f3f3finline int read(){    int x=0,f=1;char ch=getchar();    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();    return x*f;}int n,m,a[N],b[N],root[N],owo=0;struct node{    int x,lc,rc;}tree[N*40];inline void insert(int &p,int l,int r,int x){    tree[++owo]=tree[p];p=owo;tree[p].x++;    if(l==r) return;int mid=l+r>>1;    if(x<=mid) insert(tree[p].lc,l,mid,x);    else insert(tree[p].rc,mid+1,r,x);}inline int ask(int t1,int t2,int l,int r,int x){    int val=tree[tree[t2].lc].x-tree[tree[t1].lc].x;    if(l==r) return l;int mid=l+r>>1;    if(x<=val) return ask(tree[t1].lc,tree[t2].lc,l,mid,x);    else return ask(tree[t1].rc,tree[t2].rc,mid+1,r,x-val);}int main(){//  freopen("a.in","r",stdin);    n=read();m=read();    for(int i=1;i<=n;++i) a[i]=b[i]=read();    sort(b+1,b+n+1);for(int i=1;i<=n;++i) a[i]=lower_bound(b+1,b+n+1,a[i])-b;    for(int i=1;i<=n;++i){        root[i]=root[i-1];        insert(root[i],1,n,a[i]);    }while(m--){        int l=read(),r=read(),k=read();        printf("%d\n",b[ask(root[l-1],root[r],1,n,k)]);    }return 0;}