POJ 3237 Tree（树链剖分 线段树区间标记）

题意：

一棵树

三种操作

1.改变某条边的权值

2.a到b的路径取反

3,询问a到b路径上的最大值

分析：

树链剖分

线段树维护最大最小值，标记取反

代码量大

#include<cstring>
#include<string>
#include<iostream>
#include<queue>
#include<cstdio>
#include<algorithm>
#include<map>
#include<cstdlib>
#include<cmath>
#include<vector>
//#pragma comment(linker, "/STACK:1024000000,1024000000");

using namespace std;

#define INF 0x3f3f3f3f
#define maxn 20004

int x[maxn],y[maxn];
int fir[maxn],nex[maxn],v[maxn],w[maxn],e_max;
int son[maxn],fa[maxn],siz[maxn],deep[maxn],top[maxn],pos[maxn],tot;
int val1[maxn],val2[maxn],mi[4*maxn],mx[4*maxn],tag[4*maxn];

void init_()
{
    val1[0]=val1[1]=0;
    memset(fir,-1,sizeof fir);
    memset(son,-1,sizeof son);
    memset(siz,0,sizeof siz);
    e_max=0;
    tot=1;
}

void add_edge(int s,int t,int c)
{
    int e=e_max++;
    v[e]=t;
    w[e]=c;
    nex[e]=fir[s];
    fir[s]=e;
}

void dfs1(int k,int pre,int d)
{
    deep[k]=d;
    siz[k]++;
    fa[k]=pre;
    for(int i=fir[k];~i;i=nex[i])
    {
        int e=v[i];
        if(e!=pre)
        {
            val1[e]=w[i];
            dfs1(e,k,d+1);
            siz[k]+=siz[e];
            if(son[k]==-1||siz[son[k]]<siz[e]) son[k]=e;
        }
    }
}

void dfs2(int k,int sp)
{
    top[k]=sp;
    pos[k]=tot++;
    val2[pos[k]]=val1[k];
    if(son[k]==-1) return;
    dfs2(son[k],sp);
    for(int i=fir[k];~i;i=nex[i])
    {
        int e=v[i];
        if(e!=fa[k]&&e!=son[k])
        {
            dfs2(e,e);
        }
    }
}

void pushup(int k)
{
    mx[k]=max(mx[k<<1],mx[k<<1|1]);
    mi[k]=min(mi[k<<1],mi[k<<1|1]);
}

void pushdown(int k)
{
    if(!tag[k]) return ;
    tag[k<<1]^=1;
    tag[k<<1|1]^=1;
    swap(mx[k<<1],mi[k<<1]);
    mx[k<<1]=-mx[k<<1];
    mi[k<<1]=-mi[k<<1];
    swap(mx[k<<1|1],mi[k<<1|1]);
    mx[k<<1|1]=-mx[k<<1|1];
    mi[k<<1|1]=-mi[k<<1|1];
    tag[k]=0;
}

void init(int l,int r,int k)
{
    tag[k]=0;
    if(l==r)
    {
        mi[k]=mx[k]=val2[l];
        return ;
    }
    int mid=l+r>>1;
    init(l,mid,k<<1);
    init(mid+1,r,k<<1|1);
    pushup(k);
}

void update1(int d,int s,int l,int r,int k)
{
    if(s==l&&s==r)
    {
        mi[k]=mx[k]=d;
        return ;
    }
    pushdown(k);
    int mid=l+r>>1;
    if(s<=mid) update1(d,s,l,mid,k<<1);
    else update1(d,s,mid+1,r,k<<1|1);
    pushup(k);
}

void update2(int d,int s,int t,int l,int r,int k)
{
    if(s==l&&r==t)
    {
        tag[k]^=d;
        swap(mx[k],mi[k]);
        mx[k]=-mx[k];
        mi[k]=-mi[k];
        return ;
    }
    pushdown(k);
    int mid=l+r>>1;
    if(t<=mid) update2(d,s,t,l,mid,k<<1);
    else if(s>mid) update2(d,s,t,mid+1,r,k<<1|1);
    else
    {
        update2(d,s,mid,l,mid,k<<1);
        update2(d,mid+1,t,mid+1,r,k<<1|1);
    }
    pushup(k);
}

int query(int s,int t,int l,int r,int k)
{
    if(s==l&&r==t)
    {
        return mx[k];
    }
    pushdown(k);
    int mid=l+r>>1;
    if(t<=mid) return query(s,t,l,mid,k<<1);
    else if(s>mid) return query(s,t,mid+1,r,k<<1|1);
    else return max(query(s,mid,l,mid,k<<1),query(mid+1,t,mid+1,r,k<<1|1));
}

void Query(int s,int t)
{
    if(s==t)
    {
        printf("0\n");
        return ;
    }
    int ans=-INF;
    int f1=top[s],f2=top[t];
    while(f1!=f2)
    {
        if(deep[f1]<deep[f2]) swap(f1,f2),swap(s,t);
        ans=max(ans,query(pos[f1],pos[s],1,tot-1,1));
        s=fa[f1];
        f1=top[s];
    }
    if(s==t)
    {
        printf("%d\n",ans);
        return ;
    }
    if(deep[s]>deep[t]) swap(s,t);
    ans=max(ans,query(pos[s]+1,pos[t],1,tot-1,1));
    printf("%d\n",ans);
}

void Change(int s,int t)
{
    int f1=top[s],f2=top[t];
    while(f1!=f2)
    {
        if(deep[f1]<deep[f2]) swap(f1,f2),swap(s,t);
        update2(1,pos[f1],pos[s],1,tot-1,1);
        s=fa[f1];
        f1=top[s];
    }
    if(s==t) return ;
    if(deep[s]>deep[t]) swap(s,t);
    update2(1,pos[s]+1,pos[t],1,tot-1,1);
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        init_();
        int n;
        scanf("%d",&n);
        for(int i=1;i<n;i++)
        {
            int c;
            scanf("%d%d%d",&x[i],&y[i],&c);
            add_edge(x[i],y[i],c);
            add_edge(y[i],x[i],c);
        }
        dfs1(1,-1,1);
        dfs2(1,1);
        init(1,tot-1,1);
        char s[10];
        while(scanf("%s",s)!=EOF&&strcmp(s,"DONE"))
        {
            if(!strcmp(s,"QUERY"))
            {
                int l,r;
                scanf("%d%d",&l,&r);
                Query(l,r);
            }
            else if(!strcmp(s,"CHANGE"))
            {
                int i,ti;
                scanf("%d%d",&i,&ti);
                if(deep[x[i]]<deep[y[i]]) swap(x[i],y[i]);
                update1(ti,pos[x[i]],1,tot-1,1);
            }
            else
            {
                int l,r;
                scanf("%d%d",&l,&r);
                Change(l,r);
            }
        }
    }
    return 0;
}