主席树求区间第k大

主席树是可持久化线段树，维护（权值个数）线段树的前缀和。相当于对每个区间[1,i]建立n颗线段树。

我们用一个区间内的数的出现个数建线段树，所以数据大小较大时一般进行离散化。

建的是权值线段树，即用数值作为区间，每个节点存该数出现的次数，所以query返回的其实是离散后的数组b的下标idx，最终结果为b[idx]

时间复杂度分析：离散化：o(n logn)

统计并插入的复杂度是O(n logn+n logn)=O(n log⁡n)

询问的复杂度是O(n logn)

poj2104:K-th Number

theme:给定n个数，q次询问，每次求区间[l,r]中从小到大排序后第k个数。1 <= n <= 100 000, 1 <= q <= 5 000，1 <=l <=r<= n, 1 <= k <= j - i + 1

solution：主席树裸题

//theme:给定n个数，q次询问，每次求区间[l,r]中从小到大排序后第k个数。1 <= n <= 100 000, 1 <= q <= 5 000，1 <=l <=r<= n, 1 <= k <= j - i + 1
#include <iostream>
#include<cstdio>
#include<algorithm>
#define midd (l+r)/2
using namespace std;
typedef long long ll;
const int N = 200010;
int n, q, sz, cnt = 0;
int a[N], b[N], T[N];//a为原数组,b为排序离散化数组,T[i]为(值)[1,i]的根节点
int sum[N<<5], L[N<<5], R[N<<5];

inline int build(int l, int r)
{
    int rt = ++ cnt;
    sum[rt] = 0;
    if (l < r){
        L[rt] = build(l, midd);
        R[rt] = build(midd+1, r);
    }
    return rt;
}

inline int update(int pre, int l, int r, int x)
{
    int rt = ++ cnt;
    L[rt] = L[pre]; R[rt] = R[pre]; sum[rt] = sum[pre]+1;
    if (l < r){
        if (x <= midd) L[rt] = update(L[pre], l, midd, x);
        else R[rt] = update(R[pre], midd+1, r, x);
    }
    return rt;
}

inline int query(int u, int v, int l, int r, int k)//返回下标,即<=k个数
{
    if (l >= r) return l;
    int x = sum[L[v]] - sum[L[u]];
    if (x >= k) return query(L[u], L[v], l, midd, k);
    else return query(R[u], R[v], midd+1, r, k-x);
}

int main()
{
    while(scanf("%d%d", &n, &q)!=EOF){
        cnt=0;
        for (int i = 1; i <= n; i ++){
            scanf("%d", &a[i]);
            b[i] = a[i];
        }
        sort(b+1, b+1+n);
        sz = unique(b+1, b+1+n)-b-1;
        T[0] = build(1, sz);
        for (int i = 1; i <= n; i ++){
            int t = lower_bound(b+1, b+1+sz, a[i])-b;
            T[i] = update(T[i-1], 1, sz, t);
        }
        while (q --){
            int l, r, k;
            scanf("%d%d%d", &l, &r ,&k);
           // k= r-l+1-k+1;//求第k大时改一下就行
           int idx=query(T[l-1], T[r], 1, sz, k);
            int ans=b[idx];
            printf("%d\n",ans);
        }
    }
    return 0;
}