UVA 12003 Array Transformer(树套树)

#include <bits/stdc++.h>
#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>

using namespace std;

#define LL long long 
#define pii pair<int, int>
#define MP make_pair
#define ls i << 1
#define rs ls | 1
#define md (ll + rr >> 1)
#define lson ll, md, ls
#define rson md + 1, rr, rs
#define inf 0x3f3f3f3f
#define mod 1000000007
#define N 300010
#define M 400020

int readint() {
    char c;
    while((c = getchar()) && !(c >= '0' && c <= '9') && c != '-');
    int ret = c - '0', sgn = 0;
    if(c == '-') sgn = 1, ret = 0;
    while((c = getchar()) && c >= '0' && c <= '9')
        ret = ret * 10 + c - '0';
    if(sgn) ret = -ret;
    return ret;
}
int ch[N*25][2];  
int rand_val[N*25], key[N*25], sz[N*25];  
int tot, rt[N], a[N];
int creat(int val) {  
    ++tot;  
    ch[tot][0] = ch[tot][1] = 0;  
    rand_val[tot] = rand();  
    key[tot] = val;  
    sz[tot] = 1;  
    return tot;  
}
int cmp(int x, int val) {  
    if(key[x] == val) return -1;  
    return val < key[x]? 0: 1;  
}
void push_up(int x) {  
    sz[x] = sz[ch[x][0]] + sz[ch[x][1]] + 1;  
}  
void rot(int &x, int d) {  
    int k = ch[x][!d];  
    ch[x][!d] = ch[k][d];  
    ch[k][d] = x;  
    push_up(x);  
    push_up(k);  
    x = k;  
}  
void insert(int &x, int val) {  
    if(x == 0) {  
        x = creat(val);  
        return;  
    }  
    int d = cmp(x, val);  
    if(d == -1)
        d = 1;
    insert(ch[x][d], val);  
    if(rand_val[ch[x][d]] > rand_val[x]) rot(x, d ^ 1);  
    push_up(x);  
}  
void del(int &x, int val) {  
    if(x == 0) return;  
    int d = cmp(x, val);  
    if(d == -1) {  
        if(!ch[x][0]) x = ch[x][1];  
        else if(!ch[x][1]) x = ch[x][0];  
        else {  
            int d2 = rand_val[ch[x][0]] > rand_val[ch[x][1]]? 1: 0;  
            rot(x, d2);  
            del(ch[x][d2], val);  
        }  
    }  
    else del(ch[x][d], val);  
    if(x) push_up(x);  
} 
int kth(int x, int val){
    if(!x) return 0;
    int d = cmp(x, val);
    if(d <= 0)
        return kth(ch[x][0], val);
    return sz[ch[x][0]] + 1 + kth(ch[x][1], val);
}
int n;
void build(){
    for(int i = 1; i <= n; ++i)
        for(int j = i; j <= n; j += j & -j)
            insert(rt[j], a[i]);
}
int sum(int x, int v){
    int ret = 0;
    for(int i = x; i > 0; i -= i & -i)
        ret += kth(rt[i], v);
    return ret;
}
void update(int x, int pre, int cur){
    for(int i = x; i <= n; i += i & -i)
        del(rt[i], pre), insert(rt[i], cur);
}
int main(){
    int m, u;
    while(scanf("%d%d%d", &n, &m, &u) != EOF){
        tot = 0;
        for(int i = 1; i <= n; ++i){
            a[i] = readint();
            rt[i] = 0;
        }
        build();
        while(m--){
            int L, R, v, p;
            L = readint(), R = readint(), v = readint(), p = readint();
            int k = sum(R, v) - sum(L - 1, v);
            int val = 1LL * u * k / (R - L + 1);
            int pre = a[p];
            update(p, pre, val);
            a[p] = val;
        }
        for(int i = 1; i <= n; ++i)
            printf("%d\n", a[i]);
    }
    return 0;
}