链剖——BZOJ1036/Luogu2590 [ZJOI2008]树的统计

题面：BZOJ1036 Luogu2590
树剖裸题，比luogu某模板题不知道水到哪里去了。。。
今天刚入门，看到一篇很好的树剖入门教程，写得很不错：传送门
对于初学者来说讲得很详细
最后来说说这道题
单点修改直接暴力找进线段树修改
求链上最大以及求和同链上修改，只不过不必更新权值
打得谨慎一点就行了
献上我的丑陋程序+注释

#include<cstdio>#include<cstring>#include<algorithm>#include<cmath>#include<iostream>#include<cstdlib>#include<string>#include<ctime>#include<queue>#include<climits>using namespace std;char c[10];int n,nedge=0,v[100001],p[200001],nex[200001],head[200001];int fa[100001],deep[100001],s[100001],son[100001],top[100001];int sx[100001],xs[100001],ne=0;int lt[400001],rt[400001],t[400001],mt[400001];inline void addedge(int a,int b){p[++nedge]=b;nex[nedge]=head[a];head[a]=nedge;}inline void dfs(int x,int Fa,int dep){//第一次dfs求父亲,子树数目,重儿子,深度    fa[x]=Fa;deep[x]=dep;s[x]=1;    for(int k=head[x];k;k=nex[k]){        if(p[k]==Fa)continue;        dfs(p[k],x,dep+1);s[x]+=s[p[k]];        if(!son[x]||s[p[k]]>s[son[x]])son[x]=p[k];    }}inline void dfss(int x,int bh){//第二次dfs求在线段树内位置，重链最顶端位置    top[x]=bh;sx[x]=++ne;xs[sx[x]]=x;    if(!son[x])return;    dfss(son[x],bh);    for(int k=head[x];k;k=nex[k])if(p[k]!=son[x]&&p[k]!=fa[x])dfss(p[k],p[k]);}inline void build(int l,int r,int nod){//建线段树    lt[nod]=l;rt[nod]=r;    if(l==r){t[nod]=mt[nod]=v[xs[l]];return;}    int mid=l+r>>1;    build(l,mid,nod*2);build(mid+1,r,nod*2+1);    t[nod]=t[nod*2]+t[nod*2+1];mt[nod]=max(mt[nod*2],mt[nod*2+1]);}inline void xg(int x,int nod,int w){//线段树修改    if(lt[nod]==rt[nod]){t[nod]=mt[nod]=w;return;}    int mid=lt[nod]+rt[nod]>>1;    if(x<=mid)xg(x,nod*2,w);    else xg(x,nod*2+1,w);    t[nod]=t[nod*2]+t[nod*2+1];mt[nod]=max(mt[nod*2],mt[nod*2+1]);  }inline int smax(int i,int j,int nod){//线段树找最大    if(lt[nod]>=i&&rt[nod]<=j)return mt[nod];    int mid=lt[nod]+rt[nod]>>1,ans=-1e9;    if(i<=mid)ans=max(ans,smax(i,j,nod*2));    if(j>mid)ans=max(ans,smax(i,j,nod*2+1));    return ans;}inline int fmax(int x,int y){//树链上找最大    int fx=top[x],fy=top[y],ans=-1e9;    while(fx!=fy){        if(deep[fx]<deep[fy])swap(fx,fy),swap(x,y);        ans=max(ans,smax(sx[fx],sx[x],1));        x=fa[fx];fx=top[x];    }    if(deep[x]>deep[y])swap(x,y);    ans=max(ans,smax(sx[x],sx[y],1));    return ans;}inline int ssum(int i,int j,int nod){//线段树求和    if(lt[nod]>=i&&rt[nod]<=j)return t[nod];    int mid=lt[nod]+rt[nod]>>1,ans=0;    if(i<=mid)ans+=ssum(i,j,nod*2);    if(j>mid)ans+=ssum(i,j,nod*2+1);    return ans;}inline int fsum(int x,int y){//树链上求和    int fx=top[x],fy=top[y],ans=0;    while(fx!=fy){        if(deep[fx]<deep[fy])swap(fx,fy),swap(x,y);        ans+=ssum(sx[fx],sx[x],1);        x=fa[fx];fx=top[x];    }    if(deep[x]>deep[y])swap(x,y);    ans+=ssum(sx[x],sx[y],1);    return ans;}int main(){    scanf("%d",&n);    for(int i=1;i<n;i++){        int x,y;scanf("%d%d",&x,&y);        addedge(x,y);addedge(y,x);    }    for(int i=1;i<=n;i++)scanf("%d",&v[i]);    dfs(1,0,1);dfss(1,1);build(1,n,1);    int m;scanf("%d",&m);    for(int i=1;i<=m;i++){        int x,y;scanf("%s%d%d",c+1,&x,&y);        if(c[1]=='C')xg(sx[x],1,y);        if(c[2]=='M')printf("%d\n",fmax(x,y));        if(c[2]=='S')printf("%d\n",fsum(x,y));    }    return 0;}