題目描述

毛毛蟲經過及時的變形，最終逃過的一劫，離開了菜媽的菜園。 毛毛蟲經過千山萬水，歷盡千辛萬苦，最後來到了小小的紹興一中的校園裡。

爬啊爬~爬啊爬毛毛蟲爬到了一顆小小的“毛景樹”下面，發現樹上長著他最愛吃的毛毛果~ “毛景樹”上有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）處理邊權類的樹剖的時候要在修改和查詢的時候避開LCA，詳細見程式碼。

（2）線上段樹維護兩個以上的lazy標記的時候要嚴格注意傳遞的順序，這個不能寫錯，這道題的spread函式是重點。

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+100;
int n,m;

struct e {
    int u,v,w,nxt,e_id;
}edge[maxn*2];
int head[maxn];
int tot;
void addedge (int u,int v,int w,int e_id) {
    edge[tot].u=u;
    edge[tot].v=v;
    edge[tot].w=w;
    edge[tot].e_id=e_id;
    edge[tot].nxt
 
=head[u];
    head[u]=tot++;
    
    edge[tot].u=v;
    edge[tot].v=u;
    edge[tot].w=w;
    edge[tot].e_id=e_id;
    edge[tot].nxt=head[v];
    head[v]=tot++;
}
int belong[maxn];//記錄每條邊的兒子節點


int son[maxn];
int id[maxn];
int fa[maxn];
int cnt;
int dep[maxn];
int size[maxn];
int top[maxn];
int w[maxn];
int wt[maxn];

struct node {
    int l,r;
    int sum;
    int lazy1;
    int lazy2=-1;
}segTree[maxn*4];
void build (int i,int l,int r) {
    segTree[i].l=l;
    segTree[i].r=r;
    if (l==r) {
        segTree[i].sum=wt[l];
        return;
    }
    int mid=(l+r)>>1;
    build(i<<1,l,mid);
    build(i<<1|1,mid+1,r);
    segTree[i].sum=max(segTree[i<<1].sum,segTree[i<<1|1].sum);
}
void spread (int i) {
    if (segTree[i].lazy2>=0) {
        segTree[i<<1].sum=segTree[i].lazy2;
        segTree[i<<1|1].sum=segTree[i].lazy2;
        segTree[i<<1].lazy2=segTree[i].lazy2;
        segTree[i<<1|1].lazy2=segTree[i].lazy2;
        segTree[i].lazy2=-1; 
        segTree[i<<1].lazy1=0;
        segTree[i<<1|1].lazy1=0;
    }
    if (segTree[i].lazy1) {
        segTree[i<<1].sum+=segTree[i].lazy1;
        segTree[i<<1|1].sum+=segTree[i].lazy1;
        segTree[i<<1].lazy1+=segTree[i].lazy1;
        segTree[i<<1|1].lazy1+=segTree[i].lazy1;
        segTree[i].lazy1=0; 
    }
    
}
void update (int i,int l,int r,int val,int f) {
    if (l<=segTree[i].l&&segTree[i].r<=r) {
        if (f==1) {
            segTree[i].sum+=val;
            segTree[i].lazy1+=val;
            return;
        }
        else {
            segTree[i].sum=val;
            segTree[i].lazy2=val;
            segTree[i].lazy1=0;
            return;
        }
    }
    spread(i);
    int mid=(segTree[i].l+segTree[i].r)>>1;
    if (l<=mid)
        update(i<<1,l,r,val,f);
    if (r>mid)
        update(i<<1|1,l,r,val,f);
    segTree[i].sum=max(segTree[i<<1].sum,segTree[i<<1|1].sum);
}
int query (int i,int l,int r) {
    if (l<=segTree[i].l&&r>=segTree[i].r)
        return segTree[i].sum;
    spread(i);
    int mid=(segTree[i].l+segTree[i].r)>>1;
    int ans=0;
    if (l<=mid)
        ans=max(ans,query(i<<1,l,r));
    if (r>mid)
        ans=max(ans,query(i<<1|1,l,r));
    return ans;
}

int qRange (int x,int y) {
    int ans=0;
    while (top[x]!=top[y]) {
        if (dep[top[x]]<dep[top[y]]) swap(x,y);
        ans=max(ans,query(1,id[top[x]],id[x]));
        x=fa[top[x]]; 
    }
    if (dep[x]>dep[y]) swap(x,y);
    ans=max(ans,query(1,id[x]+1,id[y]));
    return ans;
}
void upRange (int x,int y,int k,int f) {
    while (top[x]!=top[y]) {
        if (dep[top[x]]<dep[top[y]]) swap(x,y);
        update(1,id[top[x]],id[x],k,f);
        x=fa[top[x]];
    }
    if (dep[x]>dep[y]) swap(x,y);
    update(1,id[x]+1,id[y],k,f);
}
int lca (int x,int y) {
    for (;top[x]!=top[y];dep[top[x]]>dep[top[y]]?x=fa[top[x]]:y=fa[top[y]]);
    return dep[x]<dep[y]?x:y;
}
void dfs1 (int x,int f,int deep) {
    dep[x]=deep;
    fa[x]=f;
    size[x]=1;
    int maxson=-1;
    for (int i=head[x];i!=-1;i=edge[i].nxt) {
        int y=edge[i].v;
        if (y==f) continue;
        belong[edge[i].e_id]=y;
        w[y]=edge[i].w;
        dfs1(y,x,deep+1);
        size[x]+=size[y];
        if (size[y]>maxson) son[x]=y,maxson=size[y];
    }
}
void dfs2 (int x,int topf) {
    id[x]=++cnt;
    wt[cnt]=w[x];
    top[x]=topf;
    if (!son[x]) return;
    dfs2(son[x],topf);
    for (int i=head[x];i!=-1;i=edge[i].nxt) {
        int y=edge[i].v;
        if (y==fa[x]||y==son[x]) continue;
        dfs2(y,y);
    }
}
 
int main () {
    scanf("%d",&n);
    for (int i=0;i<=n;i++) head[i]=-1;
    for (int i=1;i<n;i++) {
        int x,y,w;
        scanf("%d%d%d",&x,&y,&w);
        addedge(x,y,w,i);
    }
    dfs1(1,0,1);
    dfs2(1,1);
    build(1,1,n);
    while (1) {
        string s;
        cin>>s;
        if (s=="Stop") break;
        if (s=="Change") {
            int k,w;
            scanf("%d%d",&k,&w);
            update(1,id[belong[k]],id[belong[k]],w,2);
        }
        else if (s=="Cover") {
            int u,v,w;
            scanf("%d%d%d",&u,&v,&w);
            upRange(u,v,w,2);
        }
        else if (s=="Add") {
            int u,v,w;
            scanf("%d%d%d",&u,&v,&w);
            upRange(u,v,w,1);
        }
        else if (s=="Max"){
            int u,v;
            scanf("%d%d",&u,&v);
            printf("%d\n",qRange(u,v));
        }
    }
}