P4315 月下“毛景树”

题目描述

毛毛虫经过及时的变形，最终逃过的一劫，离开了菜妈的菜园。 毛毛虫经过千山万水，历尽千辛万苦，最后来到了小小的绍兴一中的校园里。

爬啊爬_{爬啊爬毛毛虫爬到了一颗小小的“毛景树”下面，发现树上长着他最爱吃的毛毛果} “毛景树”上有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之间树枝上毛毛果个数最多有多少个。

输入输出格式

输入格式：

第一行一个正整数N。

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

输出格式：

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

输入输出样例

输入样例#1：

4
1 2 8
1 3 7
3 4 9
Max 2 4
Cover 2 4 5
Add 1 4 10
Change 1 16
Max 2 4
Stop

输出样例#1：

9
16

说明

1<=N<=100,000，操作+询问数目不超过100,000。

保证在任意时刻，所有树枝上毛毛果的个数都不会超过10^9个。

【AC代码】：

const int maxn=1e5+10;
int n,a[maxn],root,head[maxn],cnt,p[maxn];
int dep[maxn],fa[maxn],siz[maxn],son[maxn];
int top[maxn],tot,in[maxn];
int t[maxn<<2],add[maxn<<2],lazy[maxn<<2];
struct node{
    int nxt,to,w;
}edge[maxn<<2];
void add_edge(int x,int y,int z){
    edge[++cnt]=(node){head[x],y,z};
    head[x]=cnt;
}
void dfs1(int x,int ffa){
    fa[x]=ffa,siz[x]=1;
    for(int i=head[x];i;i=edge[i].nxt){
        int y=edge[i].to;
        if(y!=ffa){
            dep[y]=dep[x]+1; p[y]=edge[i].w;
            dfs1(y,x); siz[x]+=siz[y];
            if(siz[y]>siz[son[x]]) son[x]=y;
        }
    }
}
void dfs2(int x,int tp){
    top[x]=tp,in[x]=++tot,a[tot]=p[x];
    if(son[x]!=0) dfs2(son[x],tp);
    for(int i=head[x];i;i=edge[i].nxt){
        int y=edge[i].to;
        if(y!=fa[x]&&y!=son[x]) dfs2(y,y);
    }
}
void build(int c,int l,int r){
    lazy[c]=-1;
    if(l==r){
        t[c]=a[l];
        return;
    }
    int mid=(l+r)>>1;
    build(c<<1,l,mid);
    build(c<<1|1,mid+1,r);
    t[c]=max(t[c<<1],t[c<<1|1]);
}
void spread(int c){
    int ls=c<<1,rs=c<<1|1;
    if(lazy[c]>=0){
        add[ls]=add[rs]=0;
        t[ls]=t[rs]=lazy[ls]=lazy[rs]=lazy[c];
        lazy[c]=-1;
    }
    if(add[c]){
        add[ls]+=add[c],add[rs]+=add[c];
        t[ls]+=add[c],t[rs]+=add[c];
        add[c]=0;
    }
}
void change1(int c,int l,int r,int x,int y,int k){
    if(x<=l&&r<=y){
        add[c]+=k; t[c]+=k;
        return;
    }
    spread(c);
    int mid=(l+r)>>1;
    if(mid>=x) change1(c<<1,l,mid,x,y,k);
    if(mid<y) change1(c<<1|1,mid+1,r,x,y,k);
    t[c]=max(t[c<<1],t[c<<1|1]);
}
void change2(int c,int l,int r,int x,int y,int k){
    if(x<=l&&r<=y){
        t[c]=lazy[c]=k;
        add[c]=0;
        return;
    }
    spread(c);
    int mid=(l+r)>>1;
    if(mid>=x) change2(c<<1,l,mid,x,y,k);
    if(mid<y) change2(c<<1|1,mid+1,r,x,y,k);
    t[c]=max(t[c<<1],t[c<<1|1]);
}
int query(int c,int l,int r,int x,int y){
    if(x<=l&&r<=y) return t[c];
    spread(c);
    int mid=(l+r)>>1,maxx=0;
    if(mid>=x) maxx=query(c<<1,l,mid,x,y);
    if(mid<y) maxx=max(maxx,query(c<<1|1,mid+1,r,x,y));
    return maxx;
}
int ask(int x,int y){
    int maxx=0;
    while(top[x]!=top[y]){
        if(dep[top[x]]<dep[top[y]]) swap(x,y);
        maxx=max(maxx,query(1,1,n,in[top[x]],in[x]));
        x=fa[top[x]];
    }
    if(dep[x]>dep[y]) swap(x,y);
    maxx=max(maxx,query(1,1,n,in[x]+1,in[y]));
    return maxx;
}
void DO1(int x,int y,int k){
    while(top[x]!=top[y]){
        if(dep[top[x]]<dep[top[y]]) swap(x,y);
        change1(1,1,n,in[top[x]],in[x],k);
        x=fa[top[x]];
    }
    if(dep[x]>dep[y]) swap(x,y);
    change1(1,1,n,in[x]+1,in[y],k);
}
void DO2(int x,int y,int k){
    while(top[x]!=top[y]){
        if(dep[top[x]]<dep[top[y]]) swap(x,y);
        change2(1,1,n,in[top[x]],in[x],k);
        x=fa[top[x]];
    }
    if(dep[x]>dep[y]) swap(x,y);
    change2(1,1,n,in[x]+1,in[y],k);
}
int main(){
    scanf("%d",&n);
    int x,y,z;
    for(int i=1;i<n;i++){
        scanf("%d%d%d",&x,&y,&z);
        add_edge(x,y,z),add_edge(y,x,z);
    }
    dep[1]=1; dfs1(1,0);
    dfs2(1,1); build(1,1,n);
    string s;
    while(1){
        cin>>s;
        if(s=="Stop") break;
        if(s=="Max"){
            scanf("%d%d",&x,&y);
            printf("%d\n",ask(x,y));
        }
        if(s=="Add"){
            scanf("%d%d%d",&x,&y,&z);
            DO1(x,y,z);
        }
        if(s=="Change"){
            scanf("%d%d",&x,&z);
            if(dep[edge[x*2-1].to]<dep[edge[x*2].to]) x=edge[x*2].to;
            else x=edge[x*2-1].to;
            change2(1,1,n,in[x],in[x],z);
        }
        if(s=="Cover"){
            scanf("%d%d%d",&x,&y,&z);
            DO2(x,y,z);
        }
    }
    return 0;
}