模板 - 主席树

像平衡树一样,只不过左子树表示的是[l,mid]的所有值,右子树表示的是[mid+1,r]的所有值,一般是维护一个cnt就可以类似在平衡树上面二分出第k小,也可以求出第k小的值在主席树中是哪个元素。当然根据你的想象力,所有只跟左子树或者右子树单侧相关的都可以找。

#include<bits/stdc++.h>
#define mid ((l+r)>>1)
using namespace std;
typedef long long ll;

const int MAXN = 200000 + 5;
int T[MAXN], tcnt;
int cnt[MAXN << 5], L[MAXN << 5], R[MAXN << 5];
ll sum[MAXN << 5];

int a[MAXN], rnk[MAXN], val[MAXN];

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

inline int update(int pre, int l, int r, int x, int val) {
    int rt = ++tcnt;
    R[rt] = R[pre];
    L[rt] = L[pre];
    cnt[rt] = cnt[pre] + 1;
    //这里传入的是离散化后的x,那应该加上x对应的原值val
    sum[rt] = sum[pre] + val;
    if(l < r) {
        if(x <= mid)
            L[rt] = update(L[pre], l, mid, x, val);
        else
            R[rt] = update(R[pre], mid + 1, r, x, val);
    }
    return rt;
}

//查询[u-1,v]中排名为rk的数
inline int query0(int u, int v, int l, int r, int rk) {
    //比整个区间的数都多,那是INF
    if(rk>cnt[v]-cnt[u])
        exit(-1);
    //直到进入一个叶子
    while(l < r) {
        int tmp=cnt[L[v]] - cnt[L[u]];
        if(tmp < rk) {
            //整个左子树加起来的数量都是<rk,所以这个排名在右子树中
            rk-=tmp;
            u = R[u], v = R[v], l = mid + 1;
        } else{
            u = L[u], v = L[v], r = mid;
        }
    }
    //最后到达了一个叶子
    return val[l] ;
}

//查询[u-1,v]中值为va的数的排名
inline int query1(int u, int v, int l, int r, int va) {
    //比整个最大的数都大,那就
    if(va>val[r])
        return cnt[v]-cnt[u]+1;
    //直到进入一个叶子
    while(l < r) {
        if(val[mid] < va) {
            //左子树的最大值比这个小,进入右子树
            u = R[u], v = R[v], l = mid + 1;
        } else{
            u = L[u], v = L[v], r = mid;
        }
    }
    //最后到达了一个叶子
    return l;
}

//查询[u-1,v]中排名不超过rk的数的个数
inline int query2(int u, int v, int l, int r, int rk) {
    int res = 0;
    while(l < r && rk < r) {
        if(mid <= rk) {
            //整个左子树都是<=rk
            res += cnt[L[v]] - cnt[L[u]];
            u = R[u], v = R[v], l = mid + 1;
        } else
            u = L[u], v = L[v], r = mid;
    }
    if(rk >= l)
        res += cnt[v] - cnt[u];
    return res ;
}

//查询[u-1,v]中排名不超过rk的数的和
inline ll query3(int u, int v, int l, int r, int rk) {
    //上面的cnt变成sum
}

//查询[u-1,v]中值不超过va的数的个数
inline int query4(int u, int v, int l, int r, int va) {
    int res = 0;
    while(l < r && va < val[r]) {
        if(val[mid] <= va) {
            //整个左子树都是<=v
            res += cnt[L[v]] - cnt[L[u]];
            u = R[u], v = R[v], l = mid + 1;
        } else
            u = L[u], v = L[v], r = mid;
    }
    if(va >= val[l])
        res += cnt[v] - cnt[u];
    return res ;
}

//查询[u-1,v]中值不超过va的数的和
inline ll query5(int u, int v, int l, int r, int va) {
    //上面的cnt变成sum
}

int main() {
#ifdef Yinku
    freopen("Yinku.in", "r", stdin);
#endif // Yinku
    int n, q;
    while(~scanf("%d%d", &n, &q)) {
        for(int i = 1; i <= n; i ++) {
            scanf("%d", &a[i]);
            rnk[i] = a[i];
        }
        sort(rnk + 1, rnk + 1 + n);
        int cb = unique(rnk + 1, rnk + 1 + n) - rnk - 1;
        tcnt = 0;
        T[0] = build(1, cb);
        for(int i = 1; i <= n; i ++) {
            int rk = lower_bound(rnk + 1, rnk + 1 + cb, a[i]) - rnk;
            val[rk] = a[i];
            T[i] = update(T[i - 1], 1, cb, rk, val[rk]);
        }
        while(q--) {
            int l, r, k;
            scanf("%d%d%d", &l, &r, &k);
            printf("%d\n", query0(T[l - 1], T[r], 1, cb, k));
        }
    }
    return 0;
}
posted @ 2019-08-16 15:43  韵意  阅读(225)  评论(0编辑  收藏  举报