【bzoj1036】树的统计[ZJOI2008]树链剖分+线段树

时间:2023-03-09 08:28:16
【bzoj1036】树的统计[ZJOI2008]树链剖分+线段树

  题目传送门:1036: [ZJOI2008]树的统计Count

  这道题是我第一次打树剖的板子,虽然代码有点长,但是“打起来很爽”,而且整道题只花了不到1.5h+,还是一遍过样例!一次提交AC!(难道前天膜cyc神犇真的起作用了?)

  按捺不住激动的心情!!!

  这道题是树剖+线段树模板题,树剖把树转化成线段后就可以单点修改+区间求和/区间求max了。

  又臭又长的93行代码:

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
struct point{
int sum,max;
}a[];
int fir[],to[],ne[];
int vis[],fa[],dep[],size[],hson[],top[],id[];
int tot,n,m,q;
char s[];
void change(int now,int l,int r,int x,int k)
{
if(l==r)a[now].sum=a[now].max=k;
else{
int mid=(l+r)>>;
if(x<=mid)change(now<<,l,mid,x,k);else change(now<<|,mid+,r,x,k);
a[now].sum=a[now<<].sum+a[now<<|].sum; a[now].max=max(a[now<<].max,a[now<<|].max);
}
}
int getsum(int now,int l,int r,int x,int y)
{
if(x<=l&&r<=y)return a[now].sum;
else{
int mid=(l+r)>>,sum=;
if(x<=mid)sum+=getsum(now<<,l,mid,x,y);
if(mid<y)sum+=getsum(now<<|,mid+,r,x,y);
return sum;
}
}
int getmax(int now,int l,int r,int x,int y)
{
if(x<=l&&r<=y)return a[now].max;
else{
int mid=(l+r)>>,mx=-<<;
if(x<=mid)mx=max(mx,getmax(now<<,l,mid,x,y));
if(mid<y)mx=max(mx,getmax(now<<|,mid+,r,x,y));
return mx;
}
}
void add(int x,int y){to[++tot]=y; ne[tot]=fir[x]; fir[x]=tot;}
void dfs1(int now,int d)
{
int i,mxson=; size[now]=; vis[now]=; dep[now]=d;
for(i=fir[now];i;i=ne[i])
if(!vis[to[i]]){
fa[to[i]]=now; dfs1(to[i],d+);
if(size[to[i]]>mxson)mxson=size[to[i]],hson[now]=to[i];
size[now]+=size[to[i]];
}
}
void dfs2(int now,int tp)
{
int i;
top[now]=tp; id[now]=++tot;
if(hson[now])dfs2(hson[now],tp);
for(i=fir[now];i;i=ne[i])
if(to[i]!=hson[now]&&to[i]!=fa[now])dfs2(to[i],to[i]);
}
int main()
{
int i,x,y;
scanf("%d",&n); tot=;
for(i=;i<n;i++)scanf("%d%d",&x,&y),add(x,y),add(y,x);
for(i=;i<=n;i++)vis[i]=; tot=;
dfs1(,); dfs2(,);
for(i=;i<=n;i++)scanf("%d",&x),change(,,n,id[i],x);
scanf("%d",&q);
for(i=;i<=q;i++){
scanf("%s%d%d",s,&x,&y);
if(s[]=='C')change(,,n,id[x],y);
else if(s[]=='M'){
int mx=-<<;
while(top[x]!=top[y]){
if(dep[top[x]]<dep[top[y]]){int tmp=x; x=y; y=tmp;}
mx=max(mx,getmax(,,n,id[top[x]],id[x])); x=fa[top[x]];
}
if(dep[x]>dep[y]){int tmp=x; x=y; y=tmp;}
printf("%d\n",max(mx,getmax(,,n,id[x],id[y])));
}
else{
int sum=;
while(top[x]!=top[y]){
if(dep[top[x]]<dep[top[y]]){int tmp=x; x=y; y=tmp;}
sum+=getsum(,,n,id[top[x]],id[x]); x=fa[top[x]];
}
if(dep[x]>dep[y]){int tmp=x; x=y; y=tmp;}
printf("%d\n",sum+getsum(,,n,id[x],id[y]));
}
}
}

树链剖分+线段树