BZOJ_P1036&Codevs_P2460 [ZJOI2008]树的统计Count(树链剖分)

BZOJ传送门 Codevs传送门
Time Limit: 10 Sec Memory Limit: 162 MB
Submit: 10826 Solved: 4376
[Submit][Status][Discuss]
Description
一棵树上有n个节点，编号分别为1到n，每个节点都有一个权值w。我们将以下面的形式来要求你对这棵树完成一些操作： I. CHANGE u t : 把结点u的权值改为t II. QMAX u v: 询问从点u到点v的路径上的节点的最大权值 III. QSUM u v: 询问从点u到点v的路径上的节点的权值和 注意：从点u到点v的路径上的节点包括u和v本身

Input
输入的第一行为一个整数n，表示节点的个数。接下来n – 1行，每行2个整数a和b，表示节点a和节点b之间有一条边相连。接下来n行，每行一个整数，第i行的整数wi表示节点i的权值。接下来1行，为一个整数q，表示操作的总数。接下来q行，每行一个操作，以“CHANGE u t”或者“QMAX u v”或者“QSUM u v”的形式给出。 对于100％的数据，保证1<=n<=30000，0<=q<=200000；中途操作中保证每个节点的权值w在-30000到30000之间。

Output
对于每个“QMAX”或者“QSUM”的操作，每行输出一个整数表示要求输出的结果。

Sample Input
4
1 2
2 3
4 1
4 2 1 3
12
QMAX 3 4
QMAX 3 3
QMAX 3 2
QMAX 2 3
QSUM 3 4
QSUM 2 1
CHANGE 1 5
QMAX 3 4
CHANGE 3 6
QMAX 3 4
QMAX 2 4
QSUM 3 4

Sample Output
4
1
2
2
10
6
5
6
5
16

HINT
Source

树链剖分裸题

#include<cstdio>
#include<climits>
#include<cstring>
#include<vector>
#include<algorithm>
#include<iostream>
using namespace std;
#define N 30005
#define lc (o<<1)
#define rc ((o<<1)|1)
#define m ((l+r)>>1)
struct Edge{int to,next;}edge[N*2];
int head[N],cnt;//邻接表 
int top[N],deep[N],son[N],f[N],size[N],s[N];//top表示重链顶点,deep表示深度,son表示重儿子,f表示父节点,size表示子节点数量,s表示权值 
int w[N],p[N],pos;//w表示与父节点连边在线段树中的位置,p表示线段树中节点的原节点编号,线段树节点数 
int n;
inline void init(){
    memset(head,-1,sizeof(head));cnt=0,pos=0;
}
inline void AddEdge(int fr,int to){edge[cnt].to=to,edge[cnt].next=head[fr],head[fr]=cnt++;}
inline void dfs1(int u,int pre,int d){//第一遍求f,deep,son,size 
    f[u]=pre,deep[u]=d,size[u]=1;
    for(int i=head[u],v=edge[i].to;i!=-1;i=edge[i].next,v=edge[i].to)
        if(v!=pre){
            dfs1(v,u,d+1);size[u]+=size[v];
            if(!son[u]||size[v]>size[son[u]]) son[u]=v;
        }
}
inline void dfs2(int u,int t){//第二遍求重儿子并编号 
    top[u]=t,w[u]=++pos,p[w[u]]=u;
    if(!son[u]) return;dfs2(son[u],t);
    for(int i=head[u],v=edge[i].to;i!=-1;i=edge[i].next,v=edge[i].to)
        if(v!=son[u]&&v!=f[u]) dfs2(v,v);
}
struct Node{int sum,ma;}node[N*4];//节点
inline void updata(int o){node[o].ma=max(node[lc].ma,node[rc].ma);
    node[o].sum=node[lc].sum+node[rc].sum;}
inline void Build(int o,int l,int r){
    if(l==r){node[o].ma=node[o].sum=s[p[l]];return;}
    Build(lc,l,m);Build(rc,m+1,r);updata(o);
}
inline void Modify(int o,int l,int r,int k,int v){
    if(l==r){node[o].sum=node[o].ma=v;return;}
    if(k<=m) Modify(lc,l,m,k,v);else Modify(rc,m+1,r,k,v);
    updata(o);return;
}
int Max(int o,int l,int r,int L,int R){
    if(L<=l&&r<=R) return node[o].ma;int ans=INT_MIN;
    if(L<=m) ans=max(ans,Max(lc,l,m,L,R));if(R>m) ans=max(ans,Max(rc,m+1,r,L,R));
    return ans;
}
int QueryMax(int u,int v){
    int f1=top[u],f2=top[v],ans=INT_MIN;
    while(f1!=f2){
        if(deep[f1]<deep[f2]) swap(u,v),swap(f1,f2);
        ans=max(ans,Max(1,1,n,w[f1],w[u]));u=f[f1],f1=top[u];
    }
    ans=max(ans,(deep[u]>deep[v])?Max(1,1,n,w[v],w[u]):Max(1,1,n,w[u],w[v]));
    return ans;
}
int Sum(int o,int l,int r,int L,int R){
    if(L<=l&&r<=R) return node[o].sum;int ans=0;
    if(L<=m) ans+=Sum(lc,l,m,L,R);if(R>m) ans+=Sum(rc,m+1,r,L,R);
    return ans;
}
int QuerySum(int u,int v){
    int f1=top[u],f2=top[v],ans=0;
    while(f1!=f2){
        if(deep[f1]<deep[f2]) swap(u,v),swap(f1,f2);
        ans+=Sum(1,1,n,w[f1],w[u]);u=f[f1],f1=top[u];
    }
    ans+=(deep[u]>deep[v])?Sum(1,1,n,w[v],w[u]):Sum(1,1,n,w[u],w[v]);
    return ans;
}
int in(){
    int x=0,v;char ch=getchar();while((ch!='-')&&(ch>'9'||ch<'0')) ch=getchar();
    if(ch=='-') v=-1,ch=getchar();else v=1;while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*v;
}
int main(){
    n=in();int v,u;init();
    for(register int i=1;i<n;i++)u=in(),v=in(),AddEdge(u,v),AddEdge(v,u);
    for(register int i=1;i<=n;i++)s[i]=in();
    dfs1(1,1,1);dfs2(1,1);Build(1,1,n);
    char opt[10];int q;q=in();memset(opt,0,sizeof(opt));
    while(q--){
        scanf("%s",opt);u=in(),v=in();
        if(opt[0]=='C') Modify(1,1,n,w[u],v);
        else if(opt[1]=='S') printf("%d\n",QuerySum(u,v));
        else printf("%d\n",QueryMax(u,v));
    }
    return 0;
}