题解:裸的,没什么好说的。
树链剖分不会的先理解一下重链轻链,然后直接扒我代码理解就行了。
贴代码:
#include <cstdio> #include <cstring> #include <iostream> #include <algorithm> #define N 30100 #define inf 0x3f3f3f3f using namespace std; struct KSD { int u,v,next; }e[N<<1]; int head[N],cnt; void add(int u,int v) { ++cnt; e[cnt].u=u; e[cnt].v=v; e[cnt].next=head[u]; head[u]=cnt; } int n,m; int w[N],tw[N]; struct Segment_Tree { int l,r,x[2]; }s[N<<2]; int deep[N],pos[N],fa[N],son[N],top[N]; int dfs1(int x,int p) { int i,v,sum=1,ax=-inf; deep[x]=deep[p]+1; fa[x]=p; for(i=head[x];i;i=e[i].next) { v=e[i].v; if(v==p)continue; int temp=dfs1(v,x); sum+=temp; if(ax<temp)ax=temp,son[x]=v; } return sum; } void dfs2(int x,int p,int r) { int i,v; pos[x]=++cnt; top[x]=r; if(son[x])dfs2(son[x],x,r); for(i=head[x];i;i=e[i].next) { v=e[i].v; if(v!=son[x]&&v!=p)dfs2(v,x,v); } } void pushup(int x) { s[x].x[0]=s[x<<1].x[0]+s[x<<1|1].x[0]; s[x].x[1]=max(s[x<<1].x[1],s[x<<1|1].x[1]); } void build(int note,int l,int r) { s[note].l=l,s[note].r=r; if(l==r) { s[note].x[0]=s[note].x[1]=w[l]; return ; } int mid=l+r>>1; build(note<<1,l,mid); build(note<<1|1,mid+1,r); pushup(note); } void change(int note,int x,int y) { if(s[note].l==s[note].r) { s[note].x[0]=s[note].x[1]=y; return ; } int mid=s[note].l+s[note].r>>1; if(x<=mid)change(note<<1,x,y); else change(note<<1|1,x,y); pushup(note); } int QUERY(int note,int l,int r,int c) { if(s[note].l==l&&r==s[note].r)return s[note].x[c]; int mid=s[note].l+s[note].r>>1; if(r<=mid)return QUERY(note<<1,l,r,c); else if(l>mid)return QUERY(note<<1|1,l,r,c); else { if(c==0)return QUERY(note<<1,l,mid,0)+QUERY(note<<1|1,mid+1,r,0); else return max(QUERY(note<<1,l,mid,1),QUERY(note<<1|1,mid+1,r,1)); } } int query(int a,int b,int c) { int A,B,ans[2]; ans[0]=0,ans[1]=-inf; for(A=top[a],B=top[b];A!=B;a=fa[A],A=top[a]) { if(deep[A]<deep[B])swap(A,B),swap(a,b); if(c==0)ans[0]+=QUERY(1,pos[A],pos[a],0); else ans[1]=max(ans[1],QUERY(1,pos[A],pos[a],1)); }if(deep[a]<deep[b])swap(a,b); if(c==0)return ans[0]+QUERY(1,pos[b],pos[a],0); else return max(ans[1],QUERY(1,pos[b],pos[a],1)); } char opt[10]; int main() { // freopen("test.in","r",stdin); // freopen("my.out","w",stdout); int i,a,b; scanf("%d",&n); for(i=1;i<n;i++) { scanf("%d%d",&a,&b); add(a,b); add(b,a); } cnt=0; dfs1(1,0); dfs2(1,0,1); for(i=1;i<=n;i++)scanf("%d",&tw[i]),w[pos[i]]=tw[i]; build(1,1,n); scanf("%d",&m); for(i=1;i<=m;i++) { scanf("%s",opt); scanf("%d%d",&a,&b); if(opt[0]=='Q') { if(opt[1]=='S')printf("%d\n",query(a,b,0)); else printf("%d\n",query(a,b,1)); } else change(1,pos[a],b); } // fclose(stdin); // fclose(stdout); return 0; }
时间: 2024-10-11 01:55:27