Cod sursă (job #465950)

#include <bits/stdc++.h>
using namespace std;

#define N 200010
int n, a[N], q, sz[N], pnod[N], mxc[N], chain[N], nrl = 1, cnt[N], ord[N], k, fw[N], bgn[N], stotal = 0;
vector <int> v[N];
FILE *fin;

void update(int pos, int val){
    for (; pos <= n; pos += (pos & -pos)) fw[pos] += val;
}

int get(int pos){
    int rs = 0;
    for (; pos; pos -= (pos & -pos)) rs += fw[pos];
    return rs;
}

int query(int x, int y){
    return (get(y) - get(x - 1));
}

int getsz(int q, int pr){
    sz[q] = 1;
    int mx = 0, idx;
    for (auto it: v[q]){
        if (it == pr) continue;
        pnod[it] = q;
        int tt = getsz(it, q);
        if (tt > mx) mx = tt, idx = it;
        sz[q] += tt;
    }
    if (mx) mxc[q] = idx;
    return sz[q];
}

void hld(int q, int pr){
    ++k;
    chain[q] = nrl;
    ord[q] = ++cnt[nrl];
    if (ord[q] == 1) bgn[nrl] = k;
    update(k, a[q]);
    if (mxc[q]) hld(mxc[q], q);
    for (auto it: v[q]){
        if (it == pr || it == mxc[q]) continue;
        nrl++; hld(it, q);
    }
}

int ask(int id){
    int aux = 0, mx = 0;
    for (auto it: v[id]){
        if (it == pnod[id]) continue;
        int st = bgn[chain[it]] + ord[it] - 1;
        int tt = query(st, st + sz[it]-1);
        if (tt > mx) mx = tt;
        aux += tt;
    }
    if (stotal-aux -a[id] > mx) mx = stotal - aux -a[id];
    return mx;
}

int main(){
    FILE *fout;
    fin = fopen("rafaela.in", "r");
    fout = fopen("rafaela.out", "w");
    fscanf(fin, "%d %d", &n, &q);
    for (int i=1, x, y; i<n; i++){
        fscanf(fin, "%d %d", &x, &y);
        v[x].push_back(y);
        v[y].push_back(x);
    }
    for (int i=1; i<=n; i++) fscanf(fin, "%d", &a[i]), stotal += a[i];
    getsz(1, -1);
    hld(1, -1);
    for (int i=1, nr, id; i<=q; i++){
        char c;
        fscanf(fin, "%1s", &c);
        //fprintf(fout, "%c\n", c);
        if (c == 'U'){
            fscanf(fin, "%d %d", &nr, &id);
            //fprintf(fout, "%d %d\n", nr, id);
            update(bgn[chain[id]]+ord[id]-1, nr);
            a[id] += nr; stotal += nr;
        }
        else{
            fscanf(fin, "%d", &id);
           // fprintf(fout, "%d\n", id);
            fprintf(fout, "%d\n", ask(id));
        }
        //fprintf(fout, "%d: %d\n", i, get(n));
    }
    return 0;
}