bzoj1984 月下“毛景树”

Description

毛毛虫经过及时的变形，最终逃过的一劫，离开了菜妈的菜园。 毛毛虫经过千山万水，历尽千辛万苦，最后来到了小小的绍兴一中的校园里。爬啊爬~爬啊爬~~毛毛虫爬到了一颗小小的“毛景树”下面，发现树上长着他最爱吃的毛毛果~~~ “毛景树”上有N个节点和N-1条树枝，但节点上是没有毛毛果的，毛毛果都是长在树枝上的。但是这棵“毛景树”有着神奇的魔力，他能改变树枝上毛毛果的个数：  Change k w：将第k条树枝上毛毛果的个数改变为w个。  Cover u v w：将节点u与节点v之间的树枝上毛毛果的个数都改变为w个。  Add u v w：将节点u与节点v之间的树枝上毛毛果的个数都增加w个。 由于毛毛虫很贪，于是他会有如下询问：  Max u v：询问节点u与节点v之间树枝上毛毛果个数最多有多少个。

Input

第一行一个正整数N。 接下来N-1行，每行三个正整数Ui，Vi和Wi，第i+1行描述第i条树枝。表示第i条树枝连接节点Ui和节点Vi，树枝上有Wi个毛毛果。 接下来是操作和询问，以“Stop”结束。

Output

对于毛毛虫的每个询问操作，输出一个答案。

树链剖分+线段树维护 区间max，区间覆盖标记，区间增加量标记

#include<cstdio>
char buf[10000005],*ptr=buf-1;
inline int input(){
int x=0,c=*++ptr,f=1;
while(c>57||c<48){if(c=='-')f=-1;c=*++ptr;}
while(c>47&&c<58)x=x*10+c-48,c=*++ptr;
return x*f;
}
inline int getop(){
int c=*++ptr;
while(c<33)c=*++ptr;
return *++ptr;
}
const int N=100050;
int mx[N*2],a[N*2],lc[N*2],rc[N*2],p=0,n;
bool cov[N*2];
int es[N*2],enx[N*2],ev[N*2],e0
,v0
,ep=2;
int l,r,ans;
int fa
,dep
,son
,sz
,top
,v
,id
,ei
[2],idr
,idp=1;
inline int max(int a,int b){return a>b?a:b;}
void dwn(int w){
if(a[w]){
if(lc[w]){
a[lc[w]]+=a[w];
mx[lc[w]]+=a[w];
a[rc[w]]+=a[w];
mx[rc[w]]+=a[w];
}
a[w]=0;
}
if(cov[w]){
if(lc[w]){
cov[lc[w]]=cov[rc[w]]=1;
mx[lc[w]]=mx[rc[w]]=mx[w];
}
cov[w]=0;
}
}
inline void up(int w){
mx[w]=max(mx[lc[w]],mx[rc[w]]);
}
int build(int L,int R){
int u=++p;
if(L==R){
mx[u]=v0[idr[L]];
return u;
}
int M=L+R>>1;
lc[u]=build(L,M);
rc[u]=build(M+1,R);
up(u);
return u;
}
void getmax(int w,int L,int R){
if(l<=L&&R<=r){
ans=max(ans,mx[w]);
return;
}
dwn(w);
int M=L+R>>1;
if(l<=M)getmax(lc[w],L,M);
if(r>M)getmax(rc[w],M+1,R);
}
void chg(int w,int L,int R){
if(l<=L&&R<=r){
a[w]=0;
cov[w]=1;
mx[w]=ans;
return;
}
dwn(w);
int M=L+R>>1;
if(l<=M)chg(lc[w],L,M);
if(r>M)chg(rc[w],M+1,R);
up(w);
}
void inc(int w,int L,int R){
if(l<=L&&R<=r){
if(!cov[w])a[w]+=ans;
mx[w]+=ans;
return;
}
dwn(w);
int M=L+R>>1;
if(l<=M)inc(lc[w],L,M);
if(r>M)inc(rc[w],M+1,R);
up(w);
}
void f1(int w,int pa){
fa[w]=pa;
dep[w]=dep[pa]+1;
sz[w]=1;
for(int i=e0[w];i;i=enx[i]){
int u=es[i];
if(u==pa)continue;
v0[u]=ev[i];
f1(u,w);
sz[w]+=sz[u];
if(sz[u]>sz[son[w]])son[w]=u;
}
}
void f2(int w,int tp){
top[w]=tp;
idr[id[w]=idp++]=w;
if(son[w])f2(son[w],tp);
for(int i=e0[w];i;i=enx[i]){
int u=es[i];
if(u!=fa[w]&&u!=son[w])f2(u,u);
}
}
void qmax(int x,int y){
ans=0;
int a=top[x],b=top[y],c;
while(a!=b){
if(dep[a]<dep[b])c=a,a=b,b=c,c=x,x=y,y=c;
l=id[a];r=id[x];
getmax(1,1,n);
x=fa[a];a=top[x];
}
if(dep[x]<dep[y])l=id[x]+1,r=id[y];
else l=id[y]+1,r=id[x];
if(l<=r)getmax(1,1,n);
printf("%d\n",ans);
}
void add(int x,int y){
int a=top[x],b=top[y],c;
while(a!=b){
if(dep[a]<dep[b])c=a,a=b,b=c,c=x,x=y,y=c;
l=id[a];r=id[x];
inc(1,1,n);
x=fa[a];a=top[x];
}
if(dep[x]<dep[y])l=id[x]+1,r=id[y];
else l=id[y]+1,r=id[x];
if(l<=r)inc(1,1,n);
}
void change(int x,int y){
int a=top[x],b=top[y],c;
while(a!=b){
if(dep[a]<dep[b])c=a,a=b,b=c,c=x,x=y,y=c;
l=id[a];r=id[x];
chg(1,1,n);
x=fa[a];a=top[x];
}
if(dep[x]<dep[y])l=id[x]+1,r=id[y];
else l=id[y]+1,r=id[x];
if(l<=r)chg(1,1,n);
}
int main(){
fread(buf,1,10000000,stdin);
n=input();
for(int i=1;i<n;i++){
int a=input(),b=input(),c=input();
es[ep]=b;enx[ep]=e0[a];ev[ep]=c;e0[a]=ep++;
es[ep]=a;enx[ep]=e0[b];ev[ep]=c;e0[b]=ep++;
ei[i][0]=a;
ei[i][1]=b;
}
f1(1,0);f2(1,1);build(1,n);
while(1){
int op=getop();
if(op=='t')break;
if(op=='a'){
int a=input(),b=input();
qmax(a,b);
}else if(op=='h'){
int a=input(),b=input();
l=r=(dep[ei[a][0]]>dep[ei[a][1]]?id[ei[a][0]]:id[ei[a][1]]);
ans=b;
chg(1,1,n);
}else if(op=='d'){
int a=input(),b=input(),c=input();
ans=c;
add(a,b);
}else{
int a=input(),b=input(),c=input();
ans=c;
change(a,b);
}
}
return 0;
}