题面: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;
}