1. 程式人生 > >樹的統計 樹鏈剖分

樹的統計 樹鏈剖分

Code:

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
typedef long long ll;
const int maxn=30000+5;
int head[maxn],nex[maxn*2],to[maxn*2];               
int dep[maxn],siz[maxn],son[maxn],p[maxn],top[maxn];
int A[maxn],maxv[maxn*4+3], val[maxn];  
int val2[maxn]
; ll sumv[maxn*4+3]; int _max; int cnt,n,cnt2; void addedge(int u,int v) { nex[++cnt]=head[u]; head[u]=cnt,to[cnt]=v; } void dfs1(int u,int fa,int cur){ p[u]=fa,dep[u]=cur,siz[u]=1; for(int i=head[u];i;i=nex[i]) { int v=to[i]; if(v!=fa) {
dfs1(v,u,cur+1); siz[u]+=siz[v]; if(son[u]==-1||siz[v]>siz[son[u]])son[u]=v; } } } void dfs2(int u,int tp) { top[u]=tp,A[u]=++cnt2; if(son[u]>0)dfs2(son[u],tp); for(int i=head[u];i;i=nex[i]) { int v=to[i]; if(v!=p[u]&&v!=son[u])dfs2(v,v); } } void build_tree(int L,
int R,int o,int arr[]) { if(L==R){ sumv[o]=maxv[o]=arr[L]; return; } int mid=(L+R)/2; build_tree(L,mid,o*2,arr); build_tree(mid+1,R,o*2+1,arr); sumv[o]=sumv[o*2]+sumv[o*2+1]; maxv[o]=max(maxv[o*2],maxv[o*2+1]); } void update(int l,int k,int L,int R,int o) { if(L==R) { sumv[o]=maxv[o]=k; return; } int mid=(L+R)/2; if(l<=mid)update(l,k,L,mid,o*2); else update(l,k,mid+1,R,o*2+1); sumv[o]=sumv[o*2]+sumv[o*2+1]; maxv[o]=max(maxv[o*2],maxv[o*2+1]); } ll query(int l,int r,int L,int R,int o) { if(l<=L&&r>=R) { _max=max(_max,maxv[o]); return sumv[o]; } int mid=(L+R)/2; ll ret=0; if(l<=mid)ret+=query(l,r,L,mid,o*2); if(r>mid)ret+=query(l,r,mid+1,R,o*2+1); return ret; } ll lca(int x,int y) { _max=-300000; ll ret=0; while(top[x]!=top[y]) { if(dep[top[x]]<dep[top[y]]){ret+=query(A[top[y]],A[y],1,n,1);y=p[top[y]];} else {ret+=query(A[top[x]],A[x],1,n,1);x=p[top[x]];} } if(dep[x]<dep[y])ret+=query(A[x],A[y],1,n,1); else ret+=query(A[y],A[x],1,n,1); return ret; } int main() { memset(son,-1,sizeof(son)); scanf("%d",&n); 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",&val[i]); dfs1(1,-1,1); //以一為根 dfs2(1,1); for(int i=1;i<=n;++i)val2[A[i]]=val[i]; build_tree(1,n,1,val2); //建樹 int T; scanf("%d",&T); while(T--) { char S[20]; scanf("%s",S); if(S[0]=='C'){ int u,t; scanf("%d%d",&u,&t); update(A[u],t,1,n,1); } if(S[1]=='M'){ int x,y; scanf("%d%d",&x,&y); lca(x,y); printf("%d\n",_max); } if(S[1]=='S') { int x,y; scanf("%d%d",&x,&y); printf("%lld\n",lca(x,y)); } } return 0; }