hdu_5221_Occupation(树剖)
题目连接：http://acm.hdu.edu.cn/showproblem.php?pid=5221
题意：给你一棵树，每个节点有一定的值，有三种操作： 1 x y 表示占领树上x-y的所有节点，2 x 表示去掉这点，3 x表示占领以这个节点为根的所有节点
题解：1 2 都是树剖操作，3则要记录树剖的时间戳，一个根节点下的点的新编号都是连续的，所以只要记录该根节点的最大时间戳就知道这个区间了
1 #include<cstdio> 2 #pragma comment(linker, "/STACK:102400000,102400000") 3 #define F(i,a,b) for(int i=a;i<=b;i++) 4 #define root 1,n,1 5 #define ls l,m,rt<<1 6 #define rs m+1,r,rt<<1|1 7 8 const int N=1e5+7; 9 int dep[N],sz[N],hs[N],top[N],tid[N],fid[N],fa[N],idx,mp[N]; 10 int t,n,m,pp,x,y,a[N],sum[N*4],vis[4*N],ans[N*4],g[N],nxt[2*N],v[2*N],ed; 11 12 inline int max(int a,int b){return a>b?a:b;} 13 inline void adg(int x,int y){v[++ed]=y,nxt[ed]=g[x],g[x]=ed;} 14 //线段树 15 void build(int l,int r,int rt){ 16 vis[rt]=ans[rt]=0; 17 if(l==r){sum[rt]=a[fid[l]];return;} 18 int m=(l+r)>>1; 19 build(ls),build(rs); 20 sum[rt]=sum[rt<<1]+sum[rt<<1|1]; 21 } 22 23 inline void pd(int rt){ 24 if(vis[rt]) 25 vis[rt<<1]=vis[rt<<1|1]=1,ans[rt<<1]=sum[rt<<1],ans[rt<<1|1]=sum[rt<<1|1],vis[rt]=0; 26 } 27 void update(int L,int R,int op,int l,int r,int rt){ 28 if(L<=l&&r<=R){ 29 vis[rt]=op,ans[rt]=op?sum[rt]:0; 30 return; 31 } 32 pd(rt); 33 int m=(l+r)>>1; 34 if(L<=m)update(L,R,op,ls); 35 if(R>m)update(L,R,op,rs); 36 ans[rt]=ans[rt<<1]+ans[rt<<1|1]; 37 } 38 //树剖 39 void dfs1(int u,int pre){ 40 dep[u]=dep[pre]+1,hs[u]=0,fa[u]=pre,sz[u]=1; 41 for(int i=g[u];i;i=nxt[i])if(v[i]!=pre) 42 dfs1(v[i],u),sz[u]+=sz[v[i]],hs[u]=(sz[v[i]]>sz[hs[u]])?v[i]:hs[u]; 43 } 44 void dfs2(int u,int tp){ 45 top[u]=tp,tid[u]=++idx,fid[idx]=u,mp[u]=idx; 46 if(hs[u])dfs2(hs[u],tp),mp[u]=max(mp[u],mp[hs[u]]); 47 for(int i=g[u];i;i=nxt[i]) 48 if(v[i]!=fa[u]&&v[i]!=hs[u])dfs2(v[i],v[i]),mp[u]=max(mp[u],mp[v[i]]); 49 } 50 51 void up(int x,int y){ 52 int fx=top[x],fy=top[y]; 53 while(fx!=fy){ 54 if(dep[fx]>dep[fy])update(tid[fx],tid[x],1,root),x=fa[fx],fx=top[x]; 55 else update(tid[fy],tid[y],1,root),y=fa[fy],fy=top[y]; 56 } 57 if(dep[x]>dep[y])x^=y,y^=x,x^=y; 58 update(tid[x],tid[y],1,root); 59 } 60 61 int main(){ 62 scanf("%d",&t); 63 while(t--){ 64 scanf("%d",&n); 65 F(i,1,n)scanf("%d",a+i); 66 F(i,1,n)g[i]=0;ed=0; 67 F(i,1,n-1)scanf("%d%d",&x,&y),adg(x,y),adg(y,x); 68 dfs1(1,0),idx=0,dfs2(1,1),build(root); 69 scanf("%d",&m); 70 F(i,1,m){ 71 scanf("%d",&pp); 72 if(pp==1)scanf("%d%d",&x,&y),up(x,y); 73 else if(pp==2)scanf("%d",&x),update(tid[x],tid[x],0,root); 74 else scanf("%d",&x),update(tid[x],mp[x],1,root); 75 printf("%d\n",ans[1]); 76 } 77 } 78 return 0; 79 }
