SPOJ 375 (樹鏈剖分+線段樹)

#include<stdio.h>#include <iostream>#include <string.h>using namespace std;const int N=210000;const int inf=0x3fffffff;int son[N],father[N],sz[N],head[N],num,ti[N],idx,dep[N],top[N];struct edge{int ed,next;}e[N*3];struct Edge{int x,y,w;}E[N*2];int max(int a,int b){if(a>b)return a;return b;}void addedge(int x,int y){e[num].ed=y;e[num].next=head[x];head[x]=num++;e[num].ed=x;e[num].next=head[y];head[y]=num++;}//****************************樹鏈部分***********************//siz[u]u的子節點個數//top[u]u所在鏈頂點//father[u]表示u的父節點//son[u]與u在同重鏈上的兒子節點//ti[u]表示u與其父親節點的連邊，線上段樹中的位置void dfs_find(int u,int fa){int i,v;sz[u]=1;dep[u]=dep[fa]+1;son[u]=0;father[u]=fa;for(i=head[u];i!=-1;i=e[i].next){v=e[i].ed;if(v==fa)continue;dfs_find(v,u);sz[u]+=sz[v];if(sz[son[u]]<sz[v])son[u]=v;}}void dfs_time(int u,int fa){int i,v;ti[u]=idx++;top[u]=fa;if(son[u]!=0)dfs_time(son[u],top[u]);for(i=head[u];i!=-1;i=e[i].next){v=e[i].ed;if(v==father[u]||v==son[u])continue;dfs_time(e[i].ed,e[i].ed);//該鏈的頂點就是該點}}//***************************線段樹部分*************************struct tree{int L,R,w;}T[N<<2];void buildTree(int L,int R,int id){T[id].L=L;T[id].R=R;T[id].w=-inf;if(L==R) return; int mid=(L+R)>>1;buildTree(L,mid,id*2);buildTree(mid+1,R,id*2+1);}void update(int id,int cp,int w){if(T[id].L==T[id].R){T[id].w=w;return;}int mid=(T[id].L+T[id].R)>>1;if(mid>=cp)update(id*2,cp,w);else update(id*2+1,cp,w);T[id].w=max(T[id*2].w,T[id*2+1].w);}int query(int L,int R,int id){if(T[id].R==R&&T[id].L==L)return T[id].w;int mid=(T[id].R+T[id].L)>>1;if(mid>=R)return query(L,R,id*2);else if(mid<L)return query(L,R,id*2+1);elsereturn max(query(L,mid,id*2),query(mid+1,R,id*2+1));}int lca(int x,int y){int ans=-inf;while(top[x]!=top[y]){if(dep[top[x]]<dep[top[y]])swap(x,y);ans=max(ans,query(ti[top[x]],ti[x],1));//x到所在鏈的頂點所有邊的最大值x=father[top[x]];//x所在鏈的頂點的父節點，轉到另一條鏈上}if(dep[x]>dep[y])swap(x,y);if(x!=y)ans=max(ans,query(ti[x]+1,ti[y],1));//ti[x]指的是x與其父親的邊，所以+1return ans;}int main(){int i,n,t,x,y,w;char str[100];scanf("%d",&t);while(t--){memset(head,-1,sizeof(head));num=0;scanf("%d",&n);for(i=1;i<n;i++){scanf("%d%d%d",&E[i].x,&E[i].y,&E[i].w);addedge(E[i].x,E[i].y);}dep[1]=0;sz[0]=0;idx=1;dfs_find(1,1);dfs_time(1,1);buildTree(2,n,1);for(i=1;i<n;i++){if(dep[E[i].x]<dep[E[i].y])swap(E[i].x,E[i].y);update(1,ti[E[i].x],E[i].w);}while(true){scanf("%s",str);if(str[0]=='D')break;else if(str[0]=='Q'){scanf("%d%d",&x,&y);printf("%d\n",lca(x,y));}else if(str[0]=='C'){scanf("%d%d",&i,&w);update(1,ti[E[i].x],w);}}printf("\n");}return 0;}