Codeforces 431E Chemistry Experiment 线段树 + 二分

Chemistry Experiment

维护一个权值线段树，然后二分答案。

#include<bits/stdc++.h>
#define LL long long
#define LD long double
#define ull unsigned 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 ALL(x) (x).begin(), (x).end()

using namespace std;

const int N = 2e5 + ;
const int inf = 0x3f3f3f3f;
const LL INF = 0x3f3f3f3f3f3f3f3f;
const int mod = 1e9 + ;
const double eps = 1e-;
const double PI = acos(-);

template<class T, class S> inline void add(T& a, S b) {a += b; if(a >= mod) a -= mod;}
template<class T, class S> inline void sub(T& a, S b) {a -= b; if(a < ) a += mod;}
template<class T, class S> inline bool chkmax(T& a, S b) {return a < b ? a = b, true : false;}
template<class T, class S> inline bool chkmin(T& a, S b) {return a > b ? a = b, true : false;}

#define lson l, mid, rt << 1
#define rson mid + 1, r, rt << 1 | 1
vector<LL> oo;
int getPos(double x) {
    return upper_bound(ALL(oo), x) - oo.begin();
}
int cnt[N << ]; LL sum[N << ];
void update(int p, int op, int l, int r, int rt) {
    if(l == r) {
        cnt[rt] += op;
        sum[rt] += op * oo[p - ];
        return;
    }
    int mid = (l + r) >> ;
    if(p <= mid) update(p, op, lson);
    else update(p, op, rson);
    cnt[rt] = cnt[rt << ] + cnt[rt <<  | ];
    sum[rt] = sum[rt << ] + sum[rt <<  | ];
}

PLI query(int L, int R, int l, int r, int rt) {
    if(L > R) return mk(, );
    if(R < l || r < L) return mk(, );
    if(L <= l && r <= R) return mk(sum[rt], cnt[rt]);
    int mid = (l + r) >> ;
    PLI tmpL = query(L, R, lson);
    PLI tmpR = query(L, R, rson);
    return mk(tmpL.fi + tmpR.fi, tmpL.se + tmpR.se);
}

bool check(double tar, LL v) {
    PLI ret = query(, getPos(tar), , SZ(oo), );
    return ret.se * tar - ret.fi > v;
}

int n, q;
LL qus[N][], h[N];

int main() {
    scanf("%d%d", &n, &q);
    for(int i = ; i <= n; i++) {
        scanf("%d", &h[i]);
        oo.push_back(h[i]);
    }
    for(int i = ; i <= q; i++) {
        scanf("%d", &qus[i][]);
        if(qus[i][] == ) {
            scanf("%lld%lld", &qus[i][], &qus[i][]);
            oo.push_back(qus[i][]);
        } else {
            scanf("%lld", &qus[i][]);
        }
    }
    sort(ALL(oo));
    oo.erase(unique(ALL(oo)), oo.end());
    for(int i = ; i <= n; i++) update(getPos(h[i]), , , SZ(oo), );
    for(int i = ; i <= q; i++) {
        if(qus[i][] == ) {
            int p = qus[i][], x = qus[i][];
            update(getPos(h[p]), -, , SZ(oo), );
            h[p] = x;
            update(getPos(h[p]), , , SZ(oo), );
        } else {
            LL v = qus[i][];
            double low = , high = 1e9 + v + ;
            for(int o = ; o < ; o++) {
                double mid = (low + high) / ;
                if(check(mid, v)) high = mid;
                else low = mid;
            }
            printf("%.15f\n", (low + high) / );
        }
    }
    return ;
}

/*
*/