BZOJ 2157 旅行(树链剖分码农题)

写了5KB，1发AC。。。

题意:给出一颗树，支持5种操作。

1.修改某条边的权值。2.将u到v的经过的边的权值取负。3.求u到v的经过的边的权值总和。4.求u到v的经过的边的权值最大值。5.求u到v经过的边的权值最小值。

基于边权的树链剖分，放在线段树上变成了区间维护问题了，线段树维护4个量min，max，sum，tag就可以了。

# include <cstdio>
# include <cstring>
# include <cstdlib>
# include <iostream>
# include <vector>
# include <queue>
# include <stack>
# include <map>
# include <bitset>
# include <set>
# include <cmath>
# include <algorithm>
using namespace std;
# define lowbit(x) ((x)&(-x))
# define pi acos(-1.0)
# define eps 1e-
# define MOD
# define INF
# define mem(a,b) memset(a,b,sizeof(a))
# define FOR(i,a,n) for(int i=a; i<=n; ++i)
# define FO(i,a,n) for(int i=a; i<n; ++i)
# define bug puts("H");
# define lch p<<,l,mid
# define rch p<<|,mid+,r
# define mp make_pair
# define pb push_back
typedef pair<int,int> PII;
typedef vector<int> VI;
# pragma comment(linker, "/STACK:1024000000,1024000000")
typedef long long LL;
int Scan() {
    int x=,f=;char ch=getchar();
    while(ch<''||ch>''){if(ch=='-')f=-;ch=getchar();}
    while(ch>=''&&ch<=''){x=x*+ch-'';ch=getchar();}
    return x*f;
}
const int N=;
//Code begin...

struct Edge{int p, next;}edge[N<<];
struct Seg{int sum, ma, mi;}seg[N<<];
bool tag[N<<];
int head[N], cnt=, E[N][], n;
int top[N], fa[N], deep[N], num[N], p[N], fp[N], son[N], pos;

void add_edge(int u, int v){
    edge[cnt].p=v; edge[cnt].next=head[u]; head[u]=cnt++;
}
void init(){mem(son,-); pos=;}
void dfs1(int u, int pre, int d){
    deep[u]=d; fa[u]=pre; num[u]=;
    for (int i=head[u]; i; i=edge[i].next) {
        int v=edge[i].p;
        if (v==pre) continue;
        dfs1(v,u,d+); num[u]+=num[v];
        if (son[u]==-||num[v]>num[son[u]]) son[u]=v;
    }
}
void getpos(int u, int sp){
    top[u]=sp; p[u]=pos++; fp[p[u]]=u;
    if (son[u]==-) return ;
    getpos(son[u],sp);
    for (int i=head[u]; i; i=edge[i].next) {
        int v=edge[i].p;
        if (v!=son[u]&&v!=fa[u]) getpos(v,v);
    }
}
void push_up(int p){
    seg[p].sum=seg[p<<].sum+seg[p<<|].sum;
    seg[p].ma=max(seg[p<<].ma,seg[p<<|].ma);
    seg[p].mi=min(seg[p<<].mi,seg[p<<|].mi);
}
void push_down(int p){
    if (!tag[p]) return ;
    seg[p].sum=-seg[p].sum; swap(seg[p].ma,seg[p].mi); seg[p].ma=-seg[p].ma; seg[p].mi=-seg[p].mi;
    tag[p<<]^=tag[p]; tag[p<<|]^=tag[p]; tag[p]=false;
}
void Update(int p, int l, int r, int L, int val){
    push_down(p);
    if (L>r||L<l) return ;
    if (L==l&&L==r) seg[p].ma=seg[p].mi=seg[p].sum=val;
    else {
        int mid=(l+r)>>;
        Update(lch,L,val); Update(rch,L,val); push_up(p);
    }
}
void Inv(int p, int l, int r, int L, int R){
    push_down(p);
    if (L>r||R<l) return ;
    if (L<=l&&R>=r) tag[p]^=, push_down(p);
    else {
        int mid=(l+r)>>;
        Inv(lch,L,R); Inv(rch,L,R); push_up(p);
    }
}
int Query_Max(int p, int l, int r, int L, int R){
    push_down(p);
    if (L>r||R<l) return -INF;
    if (L<=l&&R>=r) return seg[p].ma;
    int mid=(l+r)>>;
    return max(Query_Max(lch,L,R),Query_Max(rch,L,R));
}
int Query_Min(int p, int l, int r, int L, int R){
    push_down(p);
    if (L>r||R<l) return INF;
    if (L<=l&&R>=r) return seg[p].mi;
    int mid=(l+r)>>;
    return min(Query_Min(lch,L,R),Query_Min(rch,L,R));
}
int Query_Sum(int p, int l, int r, int L, int R){
    push_down(p);
    if (L>r||R<l) return ;
    if (L<=l&&R>=r) return seg[p].sum;
    int mid=(l+r)>>;
    return Query_Sum(lch,L,R)+Query_Sum(rch,L,R);
}
int Sol(int u, int v, int flag){
    int f1=top[u], f2=top[v], ans;
    if (flag==) ans=;
    else if (flag==) ans=-INF;
    else ans=INF;
    while (f1!=f2) {
        if (deep[f1]<deep[f2]) swap(f1,f2), swap(u,v);
        if (flag==) Inv(,,n,p[f1],p[u]);
        else if (flag==) ans+=Query_Sum(,,n,p[f1],p[u]);
        else if (flag==) ans=max(ans,Query_Max(,,n,p[f1],p[u]));
        else  ans=min(ans,Query_Min(,,n,p[f1],p[u]));
        u=fa[f1]; f1=top[u];
    }
    if (u==v) return ans;
    if (deep[u]>deep[v]) swap(u,v);
    if (flag==) {Inv(,,n,p[son[u]],p[v]); return ;}
    else if (flag==) return ans+Query_Sum(,,n,p[son[u]],p[v]);
    else if (flag==) return max(ans,Query_Max(,,n,p[son[u]],p[v]));
    else return min(ans,Query_Min(,,n,p[son[u]],p[v]));
}
int main ()
{
    int m, x, y;
    char s[];
    scanf("%d",&n); init();
    FO(i,,n) scanf("%d%d%d",&E[i][],&E[i][],&E[i][]), ++E[i][], ++E[i][], add_edge(E[i][],E[i][]), add_edge(E[i][],E[i][]);
    dfs1(,,); getpos(,);
    FO(i,,n) {
        if (deep[E[i][]]>deep[E[i][]]) swap(E[i][],E[i][]);
        Update(,,n,p[E[i][]],E[i][]);
    }
    scanf("%d",&m);
    while (m--) {
        scanf("%s%d%d",s,&x,&y);
        if (!strcmp(s,"C")) Update(,,n,p[E[x][]],y);
        else if (!strcmp(s,"N")) ++x, ++y, Sol(x,y,);
        else if (!strcmp(s,"SUM")) ++x, ++y, printf("%d\n",Sol(x,y,));
        else if (!strcmp(s,"MAX")) ++x, ++y, printf("%d\n",Sol(x,y,));
        else ++x, ++y, printf("%d\n",Sol(x,y,));
    }
    return ;
}