【ZJOI 2008】树的统计

【题目链接】

点击打开链接

【算法】

树链剖分模板题

【代码】

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

struct SegmentTree {
    int l,r,maxn,sum;
} tree[MAXN*];

int i,N,Q,u,v,t,num;
int size[MAXN+],fa[MAXN+],son[MAXN+],pos[MAXN+],
    top[MAXN+],dep[MAXN+],id[MAXN+],a[MAXN+];
string opt;
vector<int> E[MAXN+];

template <typename T> inline void read(T &x) {
    int f = ; x = ;
    char c = getchar();
    for (; !isdigit(c); c = getchar()) { if (c == '-') f = -f; }
    for (; isdigit(c); c = getchar()) x = x *  + c - '';
    x *= f;
}

template <typename T> inline void write(T x) {
    if (x < ) { putchar('-'); x = -x; }
    if (x > ) write(x/);
    putchar(x%+'');
}

template <typename T> inline void writeln(T x) {
    write(x);
    puts("");
}

inline void dfs1(int x) {
    int i,y;
    size[x] = ; son[x] = ;
    for (i = ; i < E[x].size(); i++) {
        y = E[x][i];
        if (y == fa[x]) continue;
        dep[y] = dep[x] + ;
        fa[y]= x;
        dfs1(y);
        size[x] += size[y];
        if (size[y] > size[son[x]]) son[x] = y;
    }
}

inline void dfs2(int x,int tp) {
    int i,y;
    id[x] = ++num; top[x] = tp; pos[num] = x;
    if (son[x]) dfs2(son[x],tp);
    for (i = ; i < E[x].size(); i++) {
        y = E[x][i];
        if ((y != fa[x]) && (y != son[x]))
            dfs2(y,y);
    }
}

inline void build(int index,int l,int r) {
    int mid;
    tree[index].l = l; tree[index].r = r;
    if (l == r) {
        tree[index].sum = tree[index].maxn = a[pos[l]];
        return;
    } else {
        mid = (l + r) >> ;
        build(index*,l,mid); build(index*+,mid+,r);
        tree[index].sum = tree[index*].sum + tree[index*+].sum;
        tree[index].maxn = max(tree[index*].maxn,tree[index*+].maxn);
    }
}

inline void modify(int index,int pos,int t) {
    int mid;
    if (tree[index].l == tree[index].r) {
        tree[index].sum = tree[index].maxn = t;
    } else {
        mid = (tree[index].l + tree[index].r) >> ;
        if (mid >= pos) modify(index*,pos,t);
        else modify(index*+,pos,t);
        tree[index].maxn = max(tree[index*].maxn,tree[index*+].maxn);
        tree[index].sum = tree[index*].sum + tree[index*+].sum;
    }
}

inline int query_max(int index,int l,int r) {
    int mid;
    if ((tree[index].l == l) && (tree[index].r == r)) return tree[index].maxn;
    else {
        mid = (tree[index].l + tree[index].r) >> ;
        if (mid >= r) return query_max(index*,l,r);
        else if (mid +  <= l) return query_max(index*+,l,r);
        else return max(query_max(index*,l,mid),query_max(index*+,mid+,r));
    }
}

inline int query_sum(int index,int l,int r) {
    int mid;
    if ((tree[index].l == l) && (tree[index].r == r)) return tree[index].sum;
    else {
        mid = (tree[index].l + tree[index].r) >> ;
        if (mid >= r) return query_sum(index*,l,r);
        else if (mid +  <= l) return query_sum(index*+,l,r);
        else return query_sum(index*,l,mid) + query_sum(index*+,mid+,r);
    }
}

inline int query1(int u,int v) {
    int tu = top[u],
        tv = top[v],
        ans = -2e9;
    while (tu != tv) {
        if (dep[tu] > dep[tv]) {
            swap(u,v);
            swap(tu,tv);
        }
        ans = max(ans,query_max(,id[tv],id[v]));
        v = fa[tv]; tv = top[v];
    }
    if (id[u] > id[v]) swap(u,v);
    ans = max(ans,query_max(,id[u],id[v]));
    return ans;
}

inline int query2(int u,int v) {
    int tu = top[u],
        tv = top[v],
        ans = ;
    while (tu != tv) {
        if (dep[tu] > dep[tv]) {
            swap(u,v);
            swap(tu,tv);
        }
        ans += query_sum(,id[tv],id[v]);
        v = fa[tv]; tv = top[v];
    }
    if (id[u] > id[v]) swap(u,v);
    ans += query_sum(,id[u],id[v]);
    return ans;
}

int main() {

    read(N);
    for (i = ; i < N; i++) {
        read(u); read(v);
        E[u].push_back(v);
        E[v].push_back(u);
    }
    dfs1();
    dfs2(,);
    for (i = ; i <= N; i++) read(a[i]);

    build(,,num);

    read(Q);

    while (Q--) {
        cin >> opt;
        if (opt == "CHANGE") {
            read(u); read(t);
            modify(,id[u],t);
        } else if (opt == "QMAX") {
            read(u); read(v);
            writeln(query1(u,v));
        } else {
            read(u); read(v);
            writeln(query2(u,v));
        }
    }

    return ;

}