HYSBZ 1036 【树链剖分】

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
int n,m;
int myw[30100];
bool vis[30100];
int siz[30100],dep[30100],top[30100],fa[30100],son[30100],w[30100],fp[30100];
int totw;
char typ[20];
long long INF=0x3f3f3f3f3f;
struct edge
{
    int id;
    edge *next;
};
edge edges[30100<<1];
edge *adj[30100];
struct tr
{
    int s,e,maxx;
    long long sum;
};
tr tree[30100<<2];
void build(int k,int s,int e)
{
    tree[k].s=s;
    tree[k].e=e;
    if(s==e)
    {
        tree[k].sum=tree[k].maxx=myw[fp[s]];
        return;
    }
    int mid=(s+e)>>1;
    build(k<<1,s,mid);
    build(k<<1|1,mid+1,e);
    tree[k].maxx=max(tree[k<<1].maxx,tree[k<<1|1].maxx);
    tree[k].sum=tree[k<<1|1].sum+tree[k<<1].sum;
}
int ednum;
inline void AddEdge(int a,int b)
{
    edge *tmp;
    tmp=&edges[ednum++];
    tmp->id=b;
    tmp->next=adj[a];
    adj[a]=tmp;
}

void dfs1(int id,int mdep,int f)
{
    vis[id]=1;
    dep[id]=mdep;
    siz[id]=1;
    fa[id]=f;
    son[id]=0;
    int tmpsiz=-1;
    for(edge *it=adj[id]; it; it=it->next)
    {
        if(!vis[it->id])
        {
            dfs1(it->id,mdep+1,id);
            if(tmpsiz<siz[it->id])
            {
                tmpsiz=siz[it->id];
                son[id]=it->id;
            }
            siz[id]+=siz[it->id];
        }
    }
}
void dfs2(int id,int ase)
{
    vis[id]=1;
    top[id]=ase;
    w[id]=++totw;
    fp[totw]=id;
    if(son[id])
        dfs2(son[id],ase);
    for(edge *it=adj[id]; it; it=it->next)
    {
        if(!vis[it->id])
        {
            dfs2(it->id,it->id);
        }
    }
}
long long ans;
void querySum(int k,int s,int e)
{
    if(tree[k].e==e&&tree[k].s==s)
    {
        ans+=tree[k].sum;
        return;
    }
    int mid=(tree[k].s+tree[k].e)>>1;
    if(e<=mid)
    {
        querySum(k<<1,s,e);
    }
    else if(s>mid)
    {
        querySum(k<<1|1,s,e);
    }
    else
    {
        querySum(k<<1,s,mid);
        querySum(k<<1|1,mid+1,e);
    }
}
void qsum(int a,int b)
{
    int f1=top[a],f2=top[b];

    ans=0;
    while(f1!=f2)
    {
        if(dep[f1]<dep[f2])
        {
            swap(f1,f2);
            swap(a,b);
        }
        querySum(1,w[f1],w[a]);
        a=fa[f1];
        f1=top[a];
    }
    if(dep[a]>dep[b])
        swap(a,b);
    querySum(1,w[a],w[b]);
    printf("%lld\n",ans);
}
void queryMax(int k,int s,int e)
{
    if(tree[k].e==e&&tree[k].s==s)
    {
        ans=max(ans,(long long)tree[k].maxx);
        return;
    }
    int mid=(tree[k].s+tree[k].e)>>1;
    if(e<=mid)
    {
        queryMax(k<<1,s,e);
    }
    else if(s>mid)
    {
        queryMax(k<<1|1,s,e);
    }
    else
    {
        queryMax(k<<1,s,mid);
        queryMax(k<<1|1,mid+1,e);
    }
}
void qmax(int a,int b)
{
    int f1=top[a],f2=top[b];
    ans=INF*(-1);
    while(f1!=f2)
    {
        if(dep[f1]<dep[f2])
        {
            swap(f1,f2);
            swap(a,b);
        }
        queryMax(1,w[f1],w[a]);
        a=fa[f1];
        f1=top[a];
    }
    if(dep[a]>dep[b])
        swap(a,b);
    queryMax(1,w[a],w[b]);
    printf("%lld\n",ans);
}
void update(int k,int pos,int val)
{
    if(tree[k].s==tree[k].e)
    {
        tree[k].sum=tree[k].maxx=val;
        return;
    }
    int mid=(tree[k].s+tree[k].e)>>1;
    if(pos<=mid)
        update(k<<1,pos,val);
    else
        update(k<<1|1,pos,val);
    tree[k].sum=tree[k<<1|1].sum+tree[k<<1].sum;
    tree[k].maxx=max(tree[k<<1].maxx,tree[k<<1|1].maxx);
}
int main()
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        memset(siz,0,sizeof(siz));
        memset(vis,0,sizeof(vis));
        memset(son,0,sizeof(son));
        fa[1]=0;
        totw=0;
        ednum=0;
        for(int i=1;i<=n;i++)
        {
            adj[i]=NULL;
        }
        for(int i=1; i<n; i++)
        {
            int a,b;
            scanf("%d%d",&a,&b);
            AddEdge(a,b);
            AddEdge(b,a);
        }
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&myw[i]);
        }
        dfs1(1,1,0);
        memset(vis,0,sizeof(vis));
        dfs2(1,1);
        build(1,1,totw);
        scanf("%d",&m);
        for(int i=1; i<=m; i++)
        {
            int a,b;
            scanf("%s%d%d",typ,&a,&b);
            if(typ[3]==‘X‘)
            {
                qmax(a,b);
            }
            else if(typ[3]==‘M‘)
            {
                qsum(a,b);
            }
            else if((typ[3]==‘N‘))
            {
                update(1,w[a],b);
            }
        }
    }
}
/*
8
1 2
1 4
2 3
3 8
5 6
7 5
2 5
3 7 6 5 8 2 1 0
1000
*/