树链剖分笔记
树链剖分
(不就是两个dfs吗干嘛起这么高大上的名字)
1 //hdu3966 2 #include<algorithm> 3 #include<iostream> 4 #include<cstring> 5 #include<cstdio> 6 #include<cmath> 7 #pragma comment(linker, "/STACK:1024000000,1024000000") 8 using namespace std; 9 struct edge{ 10 int v,next; 11 }a[2000001]; 12 int n,m,q,u,v,ww,now=0,tot=0,sum[2000001],cal[2000001],num[200001],head[200001],son[200001],siz[200001],dep[200001],fa[200001],top[200001],vis[200001],w[200001]; 13 char ord[10]; 14 int max(int a,int b){ 15 return a>b?a:b; 16 } 17 void add(int u,int v){ 18 a[++tot].v=v; 19 a[tot].next=head[u]; 20 head[u]=tot; 21 } 22 void dfs1(int u,int f,int dp){ 23 int v; 24 dep[u]=dp; 25 fa[u]=f; 26 siz[u]=1; 27 for(int tmp=head[u];tmp!=-1;tmp=a[tmp].next){ 28 v=a[tmp].v; 29 if(v!=f){ 30 dfs1(v,u,dp+1); 31 siz[u]+=siz[v]; 32 if(son[u]==-1||siz[v]>siz[son[u]]){ 33 son[u]=v; 34 } 35 } 36 } 37 } 38 void dfs2(int u,int tp){ 39 int v; 40 top[u]=tp; 41 vis[u]=++now; 42 w[vis[u]]=u; 43 if(son[u]==-1)return; 44 dfs2(son[u],tp); 45 for(int tmp=head[u];tmp!=-1;tmp=a[tmp].next){ 46 v=a[tmp].v; 47 if(v!=son[u]&&v!=fa[u])dfs2(v,v); 48 } 49 } 50 void pushD(int rt,int len){ 51 if(cal[rt]){ 52 cal[rt<<1]+=cal[rt]; 53 cal[rt<<1|1]+=cal[rt]; 54 sum[rt<<1]+=(len-len/2)*cal[rt]; 55 sum[rt<<1|1]+=(len/2)*cal[rt]; 56 cal[rt]=0; 57 } 58 } 59 void build(int l,int r,int rt){ 60 cal[rt]=0; 61 if(l==r){ 62 sum[rt]=num[w[l]]; 63 return; 64 } 65 int mid=(l+r)/2; 66 build(l,mid,rt<<1); 67 build(mid+1,r,rt<<1|1); 68 sum[rt]=max(sum[rt<<1],sum[rt<<1|1]); 69 } 70 void updata(int rt,int k,int l,int r,int L,int R){ 71 if(L<=l&&r<=R){ 72 cal[rt]+=k; 73 sum[rt]+=k*(r-l+1); 74 return; 75 } 76 int mid=(l+r)/2; 77 pushD(rt,r-l+1); 78 if(L<=mid)updata(rt<<1,k,l,mid,L,R); 79 if(R>mid)updata(rt<<1|1,k,mid+1,r,L,R); 80 sum[rt]=max(sum[rt<<1],sum[rt<<1|1]); 81 } 82 int query(int rt,int l,int r,int aa){ 83 if(l==r){ 84 return sum[rt]; 85 } 86 int ans=0,mid=(l+r)/2; 87 pushD(rt,r-l+1); 88 if(aa<=mid)ans=query(rt<<1,l,mid,aa); 89 else ans=query(rt<<1|1,mid+1,r,aa); 90 sum[rt]=max(sum[rt<<1],sum[rt<<1|1]); 91 return ans; 92 } 93 void update(int u,int v,int w){ 94 while(top[u]!=top[v]){ 95 if(dep[top[u]]<dep[top[v]])swap(u,v); 96 updata(1,w,1,n,vis[top[u]],vis[u]); 97 u=fa[top[u]]; 98 } 99 if(dep[u]>dep[v])swap(u,v); 100 updata(1,w,1,n,vis[u],vis[v]); 101 } 102 int main(){ 103 while(scanf("%d%d%d",&n,&m,&q)!=EOF){ 104 memset(head,255,sizeof(head)); 105 memset(son,255,sizeof(son)); 106 tot=now=0; 107 for(int i=1;i<=n;i++){ 108 scanf("%d",&num[i]); 109 } 110 for(int i=1;i<=m;i++){ 111 scanf("%d%d",&u,&v); 112 add(u,v); 113 add(v,u); 114 } 115 dfs1(1,0,0); 116 dfs2(1,1); 117 build(1,n,1); 118 for(int i=1;i<=q;i++){ 119 scanf("%s",ord); 120 if(ord[0]=='Q'){ 121 scanf("%d",&u); 122 printf("%d\n",query(1,1,n,vis[u])); 123 }else{ 124 scanf("%d%d%d",&u,&v,&ww); 125 if(ord[0]=='D')ww=-ww; 126 update(u,v,ww); 127 } 128 } 129 } 130 return 0; 131 } 132 /* 133 3 2 5 134 1 2 3 135 2 1 136 2 3 137 I 1 3 5 138 Q 2 139 D 1 2 2 140 Q 1 141 Q 3 142 -------- 143 7 144 4 145 8 146 */
