poj3237 树链剖分

WA了N发，
每次取相反数的时候，都需要交换最大值和最小值，这样才能维护动态的平衡，因为lazy标志的原因，取相反数，没有更新到底，所以如果只维护一个最大值会出问题。

#include <cstdio>
#include <cstring>
#include <algorithm>
const int N=50010;
using namespace std;
struct node
{
    int u,v,w;
}e[N];
struct d
{
    int nxt,to;
}edge[N<<1];
int cnt,tim,head[N];
int a[N];
int siz[N],son[N],fa[N],tid[N],dep[N],top[N];
void intt()
{
    tim=cnt=0;
    memset(son,-1,sizeof(son));
    memset(head,-1,sizeof(head));
}
void dfs1(int u,int pre,int de)
{
    siz[u]=1;
    dep[u]=de;
    fa[u]=pre;
    for(int i=head[u];~i;i=edge[i].nxt)
    {
        int v=edge[i].to;
        if(v!=pre)
        {
            dfs1(v,u,de+1);
            siz[u]+=siz[v];
            if(son[u]==-1||siz[son[u]]<siz[v])
                son[u]=v;
        }
    }
}
void dfs2(int u,int tp)
{
    top[u]=tp;
    tid[u]=tim++;
//    rk[tid[u]]=u;
    if(son[u]==-1) return;
    dfs2(son[u],tp);
    for(int i=head[u];~i;i=edge[i].nxt)
    {
        int v=edge[i].to;
        if(v!=son[u]&&v!=fa[u])
            dfs2(v,v);
    }
}
void add(int u,int v)
{
    edge[++cnt].to=v;edge[cnt].nxt=head[u];head[u]=cnt;
    edge[++cnt].to=u;edge[cnt].nxt=head[v];head[v]=cnt;
}


#define ls l,mid,rt<<1
#define rs mid+1,r,rt<<1|1


int sum[N<<2],sum1[N<<2],lazy[N<<2];
void changeVal(int t)
{
    int r=sum[t];
    sum[t]=-sum1[t];
    sum1[t]=-r;
}
void up(int rt)
{
    sum[rt]=max(sum[rt<<1],sum[rt<<1|1]);
    sum1[rt]=min(sum1[rt<<1],sum1[rt<<1|1]);
}
void down(int rt)
{
    if(lazy[rt])
    {
        lazy[rt<<1]^=1;
        lazy[rt<<1|1]^=1;
        changeVal(rt<<1);
        changeVal(rt<<1|1);
        lazy[rt]=0;
    }
}
void build(int l,int r,int rt)
{
    lazy[rt]=0;
    if(r==l)
    {
        sum[rt]=sum1[rt]=a[l];
        return;
    }
    int mid=(l+r)>>1;
    build(ls);
    build(rs);
    up(rt);
}
void update(int pos,int val,int l,int r,int rt)
{
    if(l==r)
    {
        sum[rt]=sum1[rt]=val;
        return;
    }
    int mid=(l+r)>>1;
    down(rt);
    if(pos<=mid) update(pos,val,ls);
    else  update(pos,val,rs);
    up(rt);
}
int query(int L,int R,int l,int r,int rt)
{
    if(L<=l&&r<=R) return sum[rt];
    int mid=(l+r)>>1;
    int ret=-0x3f3f3f3f;
    down(rt);
    if(mid>=L) ret=max(ret,query(L,R,ls));
    if(mid<R) ret=max(ret,query(L,R,rs));
    return ret;
}
void update1(int L,int R,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        lazy[rt]^=1;
        changeVal(rt);
        return;
    }
    int mid=(l+r)>>1;
    down(rt);
    if(mid>=L) update1(L,R,ls);
    if(mid<R) update1(L,R,rs);
    up(rt);
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        intt();
        int n;
        scanf("%d",&n);
        for(int i=1;i<n;i++)
        {
            scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].w);
            add(e[i].u,e[i].v);
        }
        dfs1(1,1,1);
        dfs2(1,1);
        for(int i=1;i<n;i++)
            if(dep[e[i].u]<dep[e[i].v]) a[tid[e[i].v]]=e[i].w;
            else a[tid[e[i].u]]=e[i].w;
        build(1,n-1,1);
        while(1)
        {
            char cmd[15];
            scanf("%s",cmd);
            if(cmd[0]=='D') break;
            int aa,bb;
            scanf("%d%d",&aa,&bb);
            if(cmd[0]=='Q')
            {
                int ma=-0x3f3f3f3f;
                while(top[aa]!=top[bb])
                {
                    if(dep[top[aa]]<dep[top[bb]]) swap(aa,bb);
                    ma=max(ma,query(tid[top[aa]],tid[aa],1,n-1,1));
                    aa=fa[top[aa]];
                }
                if(dep[aa]>dep[bb]) swap(aa,bb);
                if(aa!=bb) ma=max(ma,query(tid[aa]+1,tid[bb],1,n-1,1));
                printf("%d\n",ma);
            }
            else if(cmd[0]=='C')
            {
                if(dep[e[aa].u]<dep[e[aa].v]) update(tid[e[aa].v],bb,1,n-1,1);
                else update(tid[e[aa].u],bb,1,n-1,1);
            }
            else if(cmd[0]=='N')
            {
                while(top[aa]!=top[bb])
                {
                    if(dep[top[aa]]<dep[top[bb]]) swap(aa,bb);
                    update1(tid[top[aa]],tid[aa],1,n-1,1);
                    aa=fa[top[aa]];
                }
                if(dep[aa]>dep[bb]) swap(aa,bb);
                if(aa!=bb) update1(tid[aa]+1,tid[bb],1,n-1,1);
            }
        }
    }
}