POJ 2104 区间第k大 主席树（模板）

题目

题意

查询区间范围内，第 $k$ 大的数

思路

按照数列顺序依次加入，构建可持久化线段树（主席树），由于线段树的可减性，能够得到对应区间范围内的线段树
主席树的模板题了

代码

int cnt, root[maxn], a[maxn];

struct Node{
    int l, r, sum;
}T[maxn*40];

vector<int> v;
int getid(int x) {return lower_bound(v.begin(), v.end(), x) - v.begin() + 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(mid >= pos) 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) return query(l, mid, T[x].l, T[y].l, k);
    else return query(mid+1, r, T[x].r, T[y].r, k - sum);
}

int main()
{
    int n, m;
    sdd(n, m);
    rep(i, 1, n+1) {
        sd(a[i]);
        v.pb(a[i]);
    }
    sort(v.begin(), v.end()); 
    v.erase(unique(v.begin(), v.end()), v.end());
    
    rep(i, 1, n+1) update(1, n, root[i], root[i-1], getid(a[i]));
    
    rep(i, 0, m){
        int x, y, k; sddd(x, y, k);
        pd(v[query(1, n, root[x-1], root[y], k) - 1]);
    }
    
    return 0;    
}