P1505 [国家集训队]旅游

题目大意:给定一棵树,你有5种操作:1、改变其中一条边的权值,2、将节点u到节点v的所有边权取反,3、求节点u到节点v路径所有边权之和,4、求节点u到节点v路径所有边权最大值,5、求节点u到节点v路径所有边权最小值。

思路:一看题目这么长就知道码量不少qwq,这是一道裸的树链剖分,但有些细节要注意:首先是取反操作,取反后区间最大值变成区间最小值的相反数,区间最小值同理;还有就是树链剖分是解决点权的,边权怎么办呢?只要把边权变成他儿子的点权即可,其他都能用线段树解决,不过说起来简单,写起来嘛……(我才不会告诉你我写了1小时,调了1小时qwq

代码如下:压行大法吼啊

 

// luogu-judger-enable-o2
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;
const int MaxN=300000;
struct edge{
    int to,next,w;
}e[MaxN*2];
struct tree{
    int lb,rb,sum,nmax,nmin,rev;
}tr[MaxN*2];
int n,m,a[MaxN],last[MaxN],tot;
int id,dis[MaxN],fa[MaxN],son[MaxN],bel[MaxN],siz[MaxN],dfn[MaxN],top[MaxN];
inline int read() {
  char c=getchar();int num=0,f=1;
  for(;!isdigit(c);c=getchar()) if(c=='-') f=-1;
  for(;isdigit(c);c=getchar()) num=num*10+c-'0';
  return num*f;
}
void add_edge(int x,int y,int w){
    e[++tot].to=y;e[tot].next=last[x];e[tot].w=w;last[x]=tot;
    return;
}
void dfs1(int x,int f){
    dis[x]=dis[f]+1;siz[x]=1;fa[x]=f;int nmax=-0x3f3f3f3f;
    for(int i=last[x];i;i=e[i].next){
        int u=e[i].to;
        if(u==f) continue;
        dfs1(u,x);siz[x]+=siz[u];a[u]=e[i].w;
        if(siz[u]>nmax) nmax=siz[u],son[x]=u;
    }
    return;
}
void dfs2(int x,int rt){
    dfn[x]=++id;top[x]=rt;bel[id]=x;
    if(!son[x]) return;dfs2(son[x],rt);
    for(int i=last[x];i;i=e[i].next){
        int u=e[i].to;
        if(u==fa[x]||u==son[x]) continue;
        dfs2(u,u);
    }
    return;
}
void las(int x){
    tr[x].sum=tr[x<<1].sum+tr[x<<1|1].sum;
    tr[x].nmax=max(tr[x<<1].nmax,tr[x<<1|1].nmax);
    tr[x].nmin=min(tr[x<<1].nmin,tr[x<<1|1].nmin);
    return;
}
void build(int now,int l,int r){
    tr[now].lb=l;tr[now].rb=r;
    if(l==r){tr[now].sum=tr[now].nmax=tr[now].nmin=a[bel[l]];return;}
    int mid=(l+r)>>1;
    build(now<<1,l,mid);build(now<<1|1,mid+1,r);las(now);
    return;
}
void downlag(int x){
    if(tr[x].rev){
        swap(tr[x<<1].nmax,tr[x<<1].nmin);tr[x<<1].nmax*=-1;tr[x<<1].nmin*=-1;
        swap(tr[x<<1|1].nmax,tr[x<<1|1].nmin);tr[x<<1|1].nmax*=-1;tr[x<<1|1].nmin*=-1;
        tr[x<<1].sum*=-1;tr[x<<1|1].sum*=-1;
        tr[x<<1].rev^=1;tr[x<<1|1].rev^=1;tr[x].rev^=1;
    }
    return;
}
void chng(int now,int x,int w){
    if(tr[now].lb==x&&tr[now].rb==x){tr[now].sum=w;tr[now].nmax=w;tr[now].nmin=w;return;}
    downlag(now);//!
    int mid=(tr[now].lb+tr[now].rb)>>1;
    if(mid>=x) chng(now<<1,x,w);if(mid<x) chng(now<<1|1,x,w);
    las(now);return;
}
void invert(int now,int l,int r){
    if(tr[now].lb>=l&&tr[now].rb<=r){tr[now].sum*=-1;tr[now].rev^=1;swap(tr[now].nmax,tr[now].nmin);tr[now].nmax*=-1;tr[now].nmin*=-1;return;}
    downlag(now);
    int mid=(tr[now].lb+tr[now].rb)>>1;
    if(mid>=l) invert(now<<1,l,r);if(mid<r) invert(now<<1|1,l,r);
    las(now);return;
}
int query(int now,int l,int r){
    if(tr[now].lb>=l&&tr[now].rb<=r) return tr[now].sum;
    downlag(now);
    int mid=(tr[now].lb+tr[now].rb)>>1,ans=0;
    if(mid>=l) ans+=query(now<<1,l,r);if(mid<r) ans+=query(now<<1|1,l,r);
    return ans;
}
int Qmax(int now,int l,int r){
    if(tr[now].lb>=l&&tr[now].rb<=r) return tr[now].nmax;
    downlag(now);
    int mid=(tr[now].lb+tr[now].rb)>>1,ans=-0x3f3f3f3f;
    if(mid>=l) ans=max(ans,Qmax(now<<1,l,r));if(mid<r) ans=max(ans,Qmax(now<<1|1,l,r));
    return ans;
}
int Qmin(int now,int l,int r){
    if(tr[now].lb>=l&&tr[now].rb<=r) return tr[now].nmin;
    downlag(now);
    int mid=(tr[now].lb+tr[now].rb)>>1,ans=0x3f3f3f3f;
    if(mid>=l) ans=min(ans,Qmin(now<<1,l,r));if(mid<r) ans=min(ans,Qmin(now<<1|1,l,r));//Qmin->Qmax
    return ans;
}
void wvert(int x,int y){
    while(top[x]!=top[y]){
        if(dis[top[x]]<dis[top[y]]) swap(x,y);
        invert(1,dfn[top[x]],dfn[x]);x=fa[top[x]];
    }
    if(dis[x]<dis[y]) swap(x,y);
    if(y==x) return;invert(1,dfn[y]+1,dfn[x]);
    return;
}
int wmax(int x,int y){
    int ans=-0x3f3f3f3f;
    while(top[x]!=top[y]){
        if(dis[top[x]]<dis[top[y]]) swap(x,y);
        ans=max(ans,Qmax(1,dfn[top[x]],dfn[x]));x=fa[top[x]];
    }
    if(dis[x]<dis[y]) swap(x,y);
    if(x==y) return ans;ans=max(ans,Qmax(1,dfn[y]+1,dfn[x]));return ans;
} 
int wmin(int x,int y){
    int ans=0x3f3f3f3f;
    while(top[x]!=top[y]){
        if(dis[top[x]]<dis[top[y]]) swap(x,y);
        ans=min(ans,Qmin(1,dfn[top[x]],dfn[x]));x=fa[top[x]];
    }
    if(dis[x]<dis[y]) swap(x,y);
    if(x==y) return ans;ans=min(ans,Qmin(1,dfn[y]+1,dfn[x]));return ans;
}
int wsum(int x,int y){
    int ans=0;
    while(top[x]!=top[y]){
        if(dis[top[x]]<dis[top[y]]) swap(x,y);
        ans+=query(1,dfn[top[x]],dfn[x]);x=fa[top[x]];
    }
    if(dis[x]<dis[y]) swap(x,y);
    if(x==y) return ans;ans+=query(1,dfn[y]+1,dfn[x]);return ans;
}
int main(){
    int x,y,w;char c[10];
    n=read();
    for(int i=1;i<n;++i){
        x=read();y=read();w=read();++x;++y;
        add_edge(x,y,w);add_edge(y,x,w);
    }
    m=read();dfs1(1,0);dfs2(1,1);build(1,1,n);
    while(m--){
        scanf("%s",c);x=read();y=read();
        if(c[0]=='S') printf("%d\n",wsum(x+1,y+1));
        if(c[1]=='A') printf("%d\n",wmax(x+1,y+1));
        if(c[1]=='I') printf("%d\n",wmin(x+1,y+1));
        if(c[0]=='N') wvert(x+1,y+1);
        if(c[0]=='C') chng(1,dfn[x+1],y);
    }
    return 0;
}

 

posted @ 2019-07-20 19:51  X_rice  阅读(127)  评论(0编辑  收藏  举报