BZOJ1036: [ZJOI2008]树的统计Count（树链剖分）
解题思路：
树链剖分裸题，线段树维护区间和和最大值。
代码:
1 #include<cstdio> 2 #include<cstring> 3 #include<algorithm> 4 #define lll spc<<1 5 #define rrr spc<<1|1 6 typedef long long lnt; 7 using std::max; 8 using std::swap; 9 struct trnt{ 10 lnt ms; 11 lnt val; 12 }tr[1000000]; 13 struct pnt{ 14 int hd; 15 int fa; 16 int dp; 17 int tp; 18 lnt vl; 19 int ind; 20 int wgt; 21 int mxs; 22 }p[1000000]; 23 struct ent{ 24 int twd; 25 int lst; 26 }e[1000000]; 27 int cnt; 28 int n,m; 29 int dfn; 30 char cmd[100]; 31 lnt vali[1000000]; 32 void pushup(int spc) 33 { 34 tr[spc].val=tr[lll].val+tr[rrr].val; 35 tr[spc].ms=max(tr[lll].ms,tr[rrr].ms); 36 return ; 37 } 38 void Chg(int spc,lnt x) 39 { 40 tr[spc].val=x; 41 tr[spc].ms=x; 42 return ; 43 } 44 void build(int l,int r,int spc) 45 { 46 if(l==r) 47 { 48 Chg(spc,vali[l]); 49 return ; 50 } 51 int mid=(l+r)>>1; 52 build(l,mid,lll); 53 build(mid+1,r,rrr); 54 pushup(spc); 55 return ; 56 } 57 void update(int l,int r,int pos,int spc,lnt v) 58 { 59 if(l==r) 60 { 61 Chg(spc,v); 62 return ; 63 } 64 int mid=(l+r)>>1; 65 if(pos<=mid) 66 update(l,mid,pos,lll,v); 67 else 68 update(mid+1,r,pos,rrr,v); 69 pushup(spc); 70 return ; 71 } 72 lnt Max(int l,int r,int ll,int rr,int spc) 73 { 74 if(l>rr||ll>r) 75 return -0x7f3f3f3f7f7f7f7fll; 76 if(ll<=l&&r<=rr) 77 return tr[spc].ms; 78 int mid=(l+r)>>1; 79 return max(Max(l,mid,ll,rr,lll),Max(mid+1,r,ll,rr,rrr)); 80 } 81 lnt Sum(int l,int r,int ll,int rr,int spc) 82 { 83 if(l>rr||ll>r) 84 return 0ll; 85 if(ll<=l&&r<=rr) 86 return tr[spc].val; 87 int mid=(l+r)>>1; 88 return Sum(l,mid,ll,rr,lll)+Sum(mid+1,r,ll,rr,rrr); 89 } 90 void ade(int f,int t) 91 { 92 cnt++; 93 e[cnt].twd=t; 94 e[cnt].lst=p[f].hd; 95 p[f].hd=cnt; 96 return ; 97 } 98 void Basic_dfs(int x,int f) 99 { 100 p[x].fa=f; 101 p[x].dp=p[f].dp+1; 102 p[x].wgt=1; 103 int mxs=-1; 104 for(int i=p[x].hd;i;i=e[i].lst) 105 { 106 int to=e[i].twd; 107 if(to==f) 108 continue; 109 Basic_dfs(to,x); 110 p[x].wgt+=p[to].wgt; 111 if(mxs<p[to].wgt) 112 { 113 mxs=p[to].wgt; 114 p[x].mxs=to; 115 } 116 } 117 return ; 118 } 119 void Build_dfs(int x,int top) 120 { 121 if(!x) 122 return ; 123 p[x].ind=++dfn; 124 vali[dfn]=p[x].vl; 125 p[x].tp=top; 126 Build_dfs(p[x].mxs,top); 127 for(int i=p[x].hd;i;i=e[i].lst) 128 { 129 int to=e[i].twd; 130 if(p[to].ind) 131 continue; 132 Build_dfs(to,to); 133 } 134 return ; 135 } 136 int Lca(int x,int y) 137 { 138 while(p[x].tp!=p[y].tp) 139 { 140 if(p[p[x].tp].dp<p[p[y].tp].dp) 141 swap(x,y); 142 x=p[p[x].tp].fa; 143 } 144 if(p[x].dp>p[y].dp) 145 swap(x,y); 146 return x; 147 } 148 lnt Max1(int x,int y) 149 { 150 int lca=Lca(x,y); 151 lnt ans=-0x3f3f3f3f3f3f3f3fll; 152 while(p[x].tp!=p[lca].tp) 153 { 154 ans=max(ans,Max(1,dfn,p[p[x].tp].ind,p[x].ind,1)); 155 x=p[p[x].tp].fa; 156 } 157 ans=max(ans,Max(1,dfn,p[lca].ind,p[x].ind,1)); 158 while(p[y].tp!=p[lca].tp) 159 { 160 ans=max(ans,Max(1,dfn,p[p[y].tp].ind,p[y].ind,1)); 161 y=p[p[y].tp].fa; 162 } 163 ans=max(ans,Max(1,dfn,p[lca].ind,p[y].ind,1)); 164 return ans; 165 } 166 lnt Sum1(int x,int y) 167 { 168 int lca=Lca(x,y); 169 lnt ans=0; 170 while(p[x].tp!=p[lca].tp) 171 { 172 ans+=Sum(1,dfn,p[p[x].tp].ind,p[x].ind,1); 173 x=p[p[x].tp].fa; 174 } 175 ans+=Sum(1,dfn,p[lca].ind,p[x].ind,1); 176 while(p[y].tp!=p[lca].tp) 177 { 178 ans+=Sum(1,dfn,p[p[y].tp].ind,p[y].ind,1); 179 y=p[p[y].tp].fa; 180 } 181 ans+=Sum(1,dfn,p[lca].ind,p[y].ind,1); 182 ans-=p[lca].vl; 183 return ans; 184 } 185 int main() 186 { 187 scanf("%d",&n); 188 for(int i=1;i<n;i++) 189 { 190 int x,y; 191 scanf("%d%d",&x,&y); 192 ade(x,y); 193 ade(y,x); 194 } 195 for(int i=1;i<=n;i++) 196 scanf("%lld",&p[i].vl); 197 Basic_dfs(1,1); 198 Build_dfs(1,1); 199 build(1,dfn,1); 200 scanf("%d",&m); 201 while(m--) 202 { 203 scanf("%s",cmd+1); 204 if(cmd[1]=='C') 205 { 206 int x; 207 lnt y; 208 scanf("%d%lld",&x,&y); 209 p[x].vl=y; 210 update(1,dfn,p[x].ind,1,y); 211 }else if(cmd[2]=='M') 212 { 213 int x,y; 214 scanf("%d%d",&x,&y); 215 printf("%lld\n",Max1(x,y)); 216 }else{ 217 int x,y; 218 scanf("%d%d",&x,&y); 219 printf("%lld\n",Sum1(x,y)); 220 } 221 } 222 return 0; 223 }
