2019 CCPC wannfly winter camp Day 5

C - Division

思路：我们考虑到一点，从大往小取得顺序是不会有问题的，所以可以直接主席树，但是开不下空间，我们可以log分段求。

#include<bits/stdc++.h>
#define LL long long
#define fi first
#define se second
#define mk make_pair
#define PLL pair<LL, LL>
#define PLI pair<LL, int>
#define PII pair<int, int>
#define SZ(x) ((int)x.size())
#define ull unsigned long long
using namespace std;

const int N = 1e5 + ;
const int inf = 0x3f3f3f3f;
const LL INF = 0x3f3f3f3f3f3f3f3f;
const int mod = ;
const double eps = 1e-;

int n, q, all, hs[ * N], hcnt, a[N], bin[];
LL ans[ * N], sum[N], csum[N];
pair<PII, int> qus[ * N];
vector<int> vc[][N];

int root[ * N], tot;

struct node {
    LL sum;
    int cnt, ls, rs;
} o[ * N * ];

void update(int p, int l, int r, int &x, int y) {
    x = ++tot;
    o[x] = o[y];
    o[x].sum += p;
    o[x].cnt++;
    if(l == r) return;
    int mid = l + r >> ;
    if(p <= hs[mid]) update(p, l, mid, o[x].ls, o[y].ls);
    else update(p, mid + , r, o[x].rs, o[y].rs);
}

LL query(int res, int l, int r, int x, int y) {
    if(o[x].cnt - o[y].cnt <= res) return o[x].sum - o[y].sum;
    if(l == r) return 1ll * res * hs[l];
    int mid = l + r >> , cntr = o[o[x].rs].cnt - o[o[y].rs].cnt;
    if(cntr >= res) return query(res, mid + , r, o[x].rs, o[y].rs);
    else return o[o[x].rs].sum - o[o[y].rs].sum + query(res-cntr, l, mid, o[x].ls, o[y].ls);
}

int main() {
    for(int i = bin[] = ; i <= ; i++) bin[i] = bin[i - ] << ;
    scanf("%d%d", &n, &q);
    for(int i = ; i <= n; i++) scanf("%d", &a[i]), sum[i] = sum[i - ] + a[i];
    for(int i = ; i <= q; i++) {
        scanf("%d%d%d", &qus[i].fi.fi, &qus[i].fi.se, &qus[i].se);
        ans[i] = sum[qus[i].fi.se] - sum[qus[i].fi.fi - ];
    }
    for(int i = , j = ; i <= n; i++, j = ) {
        while(a[i]) {
            int val = a[i] - a[i] / ;
            while(val < bin[j]) j--;
            vc[j][i].push_back(val);
            a[i] /= ;
        }
    }
    for(int k = ; k >= ; k--) {
        tot = ; hcnt = ; all = ;
        for(int i = ; i <= n; i++) csum[i] = csum[i - ] + SZ(vc[k][i]);
        if(!csum[n]) continue;
        for(int i = ; i <= n; i++) for(auto& t : vc[k][i]) hs[++hcnt] = t;
        sort(hs + , hs +  + hcnt); hcnt = unique(hs + , hs +  + hcnt) - hs - ;
        for(int i = ; i <= n; i++) for(auto& t : vc[k][i]) update(t, , hcnt, root[all + ], root[all]), all++;
        for(int i = ; i <= q; i++) {
            if(!qus[i].se) continue;
            int L = qus[i].fi.fi, R = qus[i].fi.se, has = csum[R] - csum[L - ];
            if(has >= qus[i].se) {
                ans[i] -= query(qus[i].se, , hcnt, root[csum[R]], root[csum[L - ]]);
                qus[i].se = ;
            } else {
                ans[i] -= query(has, , hcnt, root[csum[R]], root[csum[L - ]]);
                qus[i].se -= has;
            }
        }
    }
    for(int i = ; i <= q; i++) printf("%lld\n", ans[i]);
    return ;
}

/*
*/