Luogu P1505 [国家集训队]旅游（树链剖分）

题目链接：https://www.luogu.com.cn/problem/P1505

 

 

#include<bits/stdc++.h>
using namespace std;
const int N=5e5+11,M=N<<1,Tree_Sz=3e6+11,inf=1<<29;
int n,m,a[N];
int frm[M],to[M],nxt[M],dt[M],edge,head[N];
int fa[N],dep[N],siz[N],son[N];
int seg[N],rev[N],tot,top[N];

inline int re_ad() {
    char ch=getchar(); int x=0,f=1;
    while(ch<'0' || ch>'9') { if(ch=='-') f=-1; ch=getchar(); }
    while('0'<=ch && ch<='9') x=(x<<1)+(x<<3)+ch-'0',ch=getchar();
    return x*f;
}

inline void addedge(int x,int y,int z) {
    ++edge,frm[edge]=x,to[edge]=y,nxt[edge]=head[x],dt[edge]=z,head[x]=edge;
    ++edge,frm[edge]=y,to[edge]=x,nxt[edge]=head[y],dt[edge]=z,head[y]=edge;
}

void dfs1(int d,int f,int fr) {
    fa[d]=f,dep[d]=dep[f]+1,siz[d]=1,a[d]=dt[fr];
    for(int i=head[d],u;i;i=nxt[i]) {
        u=to[i];
        if(u==f) continue;
        dfs1(u,d,i);
        siz[d]+=siz[u];
        if(siz[u]>siz[son[d]]) son[d]=u;
    }
}

void dfs2(int d) {
    if(son[d]) {
        int u=son[d];
        seg[u]=++tot,rev[tot]=u,top[u]=top[d];
        dfs2(u);
    }
    for(int i=head[d],u;i;i=nxt[i]) {
        u=to[i];
        if(u==fa[d] || u==son[d]) continue;
        seg[u]=++tot,rev[tot]=u,top[u]=u;
        dfs2(u);
    }
}

inline void print_test() {
    for(int i=1;i<=n;++i) printf("%d ",seg[i]); puts("");
    for(int i=1;i<=n;++i) printf("%d ",rev[i]); puts("");
    for(int i=1;i<=n;++i) printf("%d ",top[i]); puts("");
    for(int i=1;i<=n;++i) printf("%d ",a[i]); puts("");
}

struct Segment {
    struct Node {
        int sum,maxx,minn,tag;
        Node() { sum=tag=0,maxx=-inf,minn=inf; }
    }t[Tree_Sz];
    inline void Pushup(int d) {
//        t[d]=t[d<<1]+t[d<<1|1];
        t[d].sum=t[d<<1].sum+t[d<<1|1].sum;
        t[d].maxx=max(t[d<<1].maxx,t[d<<1|1].maxx);
        t[d].minn=min(t[d<<1].minn,t[d<<1|1].minn);
    }
    inline void Pushdown(int d,int l,int r) {
        if(!t[d].tag) return ;
        t[d<<1].tag^=t[d].tag;
        t[d<<1|1].tag^=t[d].tag;
        t[d<<1].sum=-t[d<<1].sum;
        t[d<<1|1].sum=-t[d<<1|1].sum;
        swap(t[d<<1].maxx,t[d<<1].minn);
        t[d<<1].maxx=-t[d<<1].maxx;
        t[d<<1].minn=-t[d<<1].minn;
        swap(t[d<<1|1].minn,t[d<<1|1].maxx);
        t[d<<1|1].maxx=-t[d<<1|1].maxx;
        t[d<<1|1].minn=-t[d<<1|1].minn;
        t[d].tag=0;
    }
    void build(int d,int l,int r) {
        if(l==r) { t[d].sum=t[d].maxx=t[d].minn=a[rev[l]]; return ; }
        int mid=(l+r)>>1;
        build(d<<1,l,mid),build(d<<1|1,mid+1,r);
        Pushup(d);
    }
    inline void Build() { build(1,2,n); }
    void modify(int d,int l,int r,int x,int z) {
        if(l==r) {
            t[d].sum=z*(r-l+1),t[d].maxx=z,t[d].minn=z;
            return ;
        }
        Pushdown(d,l,r);
        int mid=(l+r)>>1;
        if(x<=mid) modify(d<<1,l,mid,x,z);
        else modify(d<<1|1,mid+1,r,x,z);
        Pushup(d);
    }
    inline void Modify(int x,int z) {
        x=(dep[frm[x]]>dep[to[x]])?frm[x]:to[x];
//        printf("Change: %d %d\n",x,z);
        modify(1,2,n,seg[x],z);
    }
    void rever(int d,int l,int r,int x,int y) {
//        if(x==1634 && y==1634) printf("%d %d %d\n",d,l,r);
        if(x<=l && r<=y) {
            t[d].sum=-t[d].sum;
            swap(t[d].maxx,t[d].minn);
            t[d].maxx=-t[d].maxx;
            t[d].minn=-t[d].minn;
            t[d].tag^=1;
            return ;
        }
        Pushdown(d,l,r);
        int mid=(l+r)>>1;
        if(x<=mid) rever(d<<1,l,mid,x,y);
        if(y>=mid+1) rever(d<<1|1,mid+1,r,x,y);
        Pushup(d);
//        if(x==1634 && y==1634) printf("end: %d\n",d);
    }
    inline void Rever(int x,int y,int t) {
//        if(x==1631 && y==1631) printf("%d %d %d %d\n",seg[x],seg[y],t,n);
        rever(1,2,n,seg[x]+t,seg[y]);
    }
    Node query(int d,int l,int r,int x,int y) {
        if(x<=l && r<=y) return t[d];
        Pushdown(d,l,r);
        int mid=(l+r)>>1; Node res,tmp;
        if(x<=mid) {
            tmp=query(d<<1,l,mid,x,y);
            res.sum+=tmp.sum,res.maxx=max(res.maxx,tmp.maxx),res.minn=min(res.minn,tmp.minn);
        }
        if(y>=mid+1) {
            tmp=query(d<<1|1,mid+1,r,x,y);
            res.sum+=tmp.sum,res.maxx=max(res.maxx,tmp.maxx),res.minn=min(res.minn,tmp.minn);
        }
        return res;
    }
    inline int Query(int x,int y,int op,int t) {
        Node res;
        res=query(1,2,n,seg[x]+t,seg[y]);
        if(op==1) return res.sum;
        else if(op==2) return res.maxx;
        else return res.minn;
    }
    void search(int d,int l,int r) {
//        printf("%d %d %d %d %d %d\n",d,l,r,t[d].sum,t[d].maxx,t[d].minn);
        if(l==r) return ;
        int mid=(l+r)>>1;
        search(d<<1,l,mid),search(d<<1|1,mid+1,r);
    }
}st;

inline void Reverse(int x,int y) {
//    if(x==1631 && y==69) printf("awa\n");
    if(dep[x]<dep[y]) x^=y^=x^=y;
    int fx=top[x],fy=top[y];
    while(fx!=fy) {
//        if(x==1631 && y==69) printf("%d %d %d %d\n",x,y,fx,fy);
        if(dep[fx]<dep[fy]) x^=y^=x^=y,fx^=fy^=fx^=fy;
        st.Rever(fx,x,0);
//        if(x==1631 && y==69) printf("qwq %d %d %d %d\n",x,fx,fa[fx],top[fa[fx]]);
        x=fa[fx],fx=top[x];
    }
    if(dep[x]>dep[y]) x^=y^=x^=y;
//    printf("%d %d\n",x,y);
    if(x!=y) st.Rever(x,y,1);
}

inline void Ask(int x,int y,int op) {
    if(dep[x]<dep[y]) x^=y^=x^=y;
    int fx=top[x],fy=top[y],res;
    if(op==1) res=0;
    else if(op==2) res=-inf;
    else res=inf;
    while(fx!=fy) {
        if(dep[fx]<dep[fy]) x^=y^=x^=y,fx^=fy^=fx^=fy;
        if(op==1) res+=st.Query(fx,x,1,0);
        else if(op==2) res=max(res,st.Query(fx,x,2,0));
        else res=min(res,st.Query(fx,x,3,0));
        x=fa[fx],fx=top[x];
    }
    if(dep[x]>dep[y]) x^=y^=x^=y;
    if(x!=y) {
        if(op==1) res+=st.Query(x,y,1,1);
        else if(op==2) res=max(res,st.Query(x,y,2,1));
        else res=min(res,st.Query(x,y,3,1));
    }
    printf("%d\n",res);
}

int main()
{
//    freopen("P1505_1.in","r",stdin);
//    freopen("P1505_1.ans","w",stdout);
    n=re_ad(); edge=1;
    for(int i=1,x,y,z;i<n;++i) x=re_ad(),y=re_ad(),z=re_ad(),addedge(x+1,y+1,z);
    dfs1(1,0,0),tot=seg[1]=rev[1]=top[1]=1,dfs2(1),st.Build();
//    print_test();
    m=re_ad(); char str[11];
    for(int i=1,x,y;i<=m;++i) {
        scanf("%s",str); x=re_ad(),y=re_ad();
//        printf("%c%c %d %d\n",str[0],str[1],x,y);
        if(str[0]=='N') Reverse(x+1,y+1);
        else if(str[0]=='C') st.Modify(x*2,y);
        else if(str[0]=='S') Ask(x+1,y+1,1);
        else Ask(x+1,y+1,(str[1]=='A')?2:3);
//        st.search(1,2,n); puts("");
    }
    return 0;
}