zoukankan      html  css  js  c++  java
  • 模板

    像平衡树一样,只不过左子树表示的是[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
    ", query0(T[l - 1], T[r], 1, cb, k));
            }
        }
        return 0;
    }
    
  • 相关阅读:
    idou老师教你学Istio 08: 调用链埋点是否真的“零修改”?
    idou老师教你学Istio 07: 如何用istio实现请求超时管理
    idou老师教你学Istio06: 如何用istio实现流量迁移
    Strusts2笔记8--文件的上传和下载
    Strusts2笔记7--国际化
    Strusts2笔记6--拦截器
    Strusts2笔记5--数据验证
    Strusts2笔记4--类型转换器
    Struts2笔记3--获取ServletAPI和OGNL与值栈
    Struts2笔记2--动态方法调用和Action接收请求方式
  • 原文地址:https://www.cnblogs.com/Yinku/p/11364332.html
Copyright © 2011-2022 走看看