主席树（可持续化线段树）

Posted 2022-08-30 starve

学习粗：https://blog.csdn.net/creatorx/article/details/75446472

题：http://poj.org/problem?id=2104（静态主席树）

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define pb push_back

inline int read()
    int sum=0,x=1;
    char ch=getchar();
    while(ch<‘0‘||ch>‘9‘)
        if(ch==‘-‘)
            x=0;
        ch=getchar();
    
    while(ch>=‘0‘&&ch<=‘9‘)
        sum=(sum<<1)+(sum<<3)+(ch^48),ch=getchar();
    return x?sum:-sum;

inline void write(int x)
    if(x<0)
        putchar(‘-‘),x=-x;
    if(x>9)
        write(x/10);
    putchar(x%10+‘0‘);

const int M=1e5+5;
int n,m,cnt=1;
struct node
    int L,R,sum;
tree[M*20];
struct NODE
    int x,id;
a[M];
bool cmp(NODE p,NODE q)
    return p.x<q.x;

int root[M],Rank[M];
void update(int pos,int &root,int l,int r)
    tree[cnt++]=tree[root];
    root=cnt-1;
    tree[root].sum+=1;
    if(l==r)
        return ;
    int midd=(l+r)>>1;
    if(pos<=midd)
        update(pos,tree[root].L,l,midd);
    else
        update(pos,tree[root].R,midd+1,r);
    

int query(int i,int j,int k,int l,int r)
    int d=tree[tree[j].L].sum-tree[tree[i].L].sum;
    if(l==r)
        return l;
    int midd=(l+r)>>1;
    if(k<=d)
        return query(tree[i].L,tree[j].L,k,l,midd);
    else
        return query(tree[i].R,tree[j].R,k-d,midd+1,r);

int main()
    n=read(),m=read();
//    write(n);
//    puts("");
//    write(m);
    for(int i=1;i<=n;i++)
        a[i].x=read();
        a[i].id=i;
    
    //lisan
    sort(a+1,a+n+1,cmp);
    for(int i=1;i<=n;i++)
        Rank[a[i].id]=i;//Rank[i]=j:原序列第i个位置是第j名（按原序列升序） 
    
    for(int i=1;i<=n;i++)
        root[i]=root[i-1];
        update(Rank[i],root[i],1,n);
    
    for(int i=1;i<=m;i++)
        int l=read(),r=read(),k=read();
        write(a[query(root[l-1],root[r],k,1,n)].x);
        putchar(‘\n‘);
    
    return 0;

View Code