hdu5052 Yaoge’s maximum profit 树链剖分

一棵树上，从u走到v，在某点买入，咋之后的某点卖出，求最大利润。

维护正着走和反着走的最大利润。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
#include<vector>
#include<set>
#include<map>
#include<queue>
#include<bitset>
#define fi first
#define se second
#define pii pair<int,int>
#define ll long long
#define inf 1000000000
#define eps 1e-8
using namespace std;
const int maxn=100005;
int son[maxn],top[maxn],dep[maxn],fa[maxn],siz[maxn],id[maxn],fid[maxn];
int sz;
vector<int>g[maxn];
int val[maxn];
int n,q;
struct Node
{
    int mi,mx;
    int ans;
    int o;
    Node(int a,int b,int c,int d):mi(a),mx(b),ans(c),o(d){}
    Node(){}
}node[maxn<<2];
int add[maxn<<2];
void dfs(int u,int f)
{
    siz[u]=1;
    son[u]=0;
    for(int i=0;i<g[u].size();i++) {
        if(g[u][i]==f)
            continue;
        int v=g[u][i];
        fa[v]=u;
        dep[v]=dep[u]+1;
        dfs(v,u);
        if(siz[v]>siz[son[u]]) son[u]=v;
        siz[u]+=siz[v];
    }
}
void build(int u,int w)
{
    id[u]=++sz;
    fid[sz]=u;
    top[u]=w;
    if(son[u]) build(son[u],w);
    for(int i=0;i<g[u].size();i++) {
        int v=g[u][i];
        if(v==fa[u]) continue;
        if(v!=son[u]) build(v,v);
    }
}
void T()
{
    int rt=(n+1)/2;
    sz=0;
    dep[rt]=0;
    fa[rt]=0;
    dfs(rt,0);
    build(rt,rt);
}
void pushUp(int rt)
{
    node[rt].mx=max(node[rt<<1].mx,node[rt<<1|1].mx);
    node[rt].mi=min(node[rt<<1].mi,node[rt<<1|1].mi);
    node[rt].ans=max(node[rt<<1].ans,node[rt<<1|1].ans);
    node[rt].ans=max(node[rt].ans,node[rt<<1|1].mx-node[rt<<1].mi);
    node[rt].o=max(node[rt<<1].o,node[rt<<1|1].o);
    node[rt].o=max(node[rt].o,node[rt<<1].mx-node[rt<<1|1].mi);
}
void pushDown(int rt)
{
    if(add[rt]==0) return;
    node[rt<<1].mi+=add[rt];
    node[rt<<1].mx+=add[rt];
    node[rt<<1|1].mi+=add[rt];
    node[rt<<1|1].mx+=add[rt];
    add[rt<<1]+=add[rt];
    add[rt<<1|1]+=add[rt];
    add[rt]=0;
}
void update(int L,int R,int c,int l,int r,int rt)
{
    if(L<=l&&R>=r) {
        node[rt].mx+=c;
        node[rt].mi+=c;
        add[rt]+=c;
        return;
    }
    pushDown(rt);
    int m=((l+r)>>1);
    //Node u={inf,-inf,0};
    if(L<=m)
        update(L,R,c,l,m,rt<<1);
    if(R>m)
        update(L,R,c,m+1,r,rt<<1|1);
    pushUp(rt);
}
void update(int a,int b,int c)
{
    int f1=top[a],f2=top[b];
    while(f1!=f2) {
        if(dep[f1]<dep[f2]) {
            swap(f1,f2);
            swap(a,b);
        }
        update(id[f1],id[a],c,1,n,1);
        a=fa[f1]; f1=top[a];
    }
    if(dep[a]>dep[b]) swap(a,b);
    update(id[a],id[b],c,1,n,1);
}
Node unite(Node a,Node b)
{
    Node u;
    u.mx=max(a.mx,b.mx);
    u.mi=min(a.mi,b.mi);
    u.ans=max(a.ans,b.ans);
    u.ans=max(u.ans,b.mx-a.mi);
    u.o=max(a.o,b.o);
    u.o=max(u.o,a.mx-b.mi);
    return u;
}
Node query(int L,int R,int l,int r,int rt)
{
    if(L<=l&&R>=r)
        return node[rt];
    pushDown(rt);
    Node u=Node(inf,-inf,0,0);
    int m=((l+r)>>1);
    if(L<=m)
        u=unite(u,query(L,R,l,m,rt<<1));
    if(R>m)
        u=unite(u,query(L,R,m+1,r,rt<<1|1));
    pushUp(rt);
    return u;
}
int query(int u,int v)
{
    int a = top[u], b = top[v];
    Node lpath = Node(inf , -inf, 0,0), rpath = Node(inf, -inf, 0,0);
    while(a != b){
        if( dep[a] > dep[b]){
            Node res = query(id[a], id[u],1,n,1);
            //cout<<res.mi<<" "<<res.mx<<" "<<res.ans<<endl;
            swap(res.ans, res.o);
            lpath = unite(lpath, res);
            u = fa[a];
            a = top[u];
        }
        else{
            Node res = query(id[b], id[v],1,n,1);
            //cout<<res.mi<<" "<<res.mx<<" "<<res.ans<<endl;
            rpath = unite(res, rpath);
            v = fa[b];
            b = top[v];
        }
    }
    Node ans;
    if( dep[u] < dep[v] ){
        ans = query(id[u], id[v],1,n,1);
    }
    else{
        ans = query(id[v], id[u],1,n,1);
        swap(ans.ans,ans.o);
    }
    ans = unite(lpath, ans);
    ans = unite(ans, rpath);
    return ans.ans;
}
void buildTree(int l,int r,int rt)
{
    add[rt]=0;
    if(l==r) {
        int a=fid[l];
        node[rt].mi=node[rt].mx=val[a];
        node[rt].ans=node[rt].o=0;
        return;
    }
    int m=((l+r)>>1);
    buildTree(l,m,rt<<1);
    buildTree(m+1,r,rt<<1|1);
    pushUp(rt);
}
int main()
{
    int t;
    int a,b,c;
    scanf("%d",&t);
    while(t--) {
        scanf("%d",&n);
        for(int i=1;i<=n;i++) {
            scanf("%d",&val[i]);
            g[i].clear();
        }
        for(int i=0;i<n-1;i++) {
            scanf("%d%d",&a,&b);
            g[a].push_back(b);
            g[b].push_back(a);
        }
        T();
        buildTree(1,n,1);
        //cout<<node[1].mi<<" "<<node[1].mx<<" "<<node[1].ans<<endl;
        scanf("%d",&q);
        while(q--) {
            scanf("%d%d%d",&a,&b,&c);
            printf("%d\n",query(a,b));
            update(a,b,c);
        }
    }
    return 0;
}