树链剖分边权 poj 3237

#include <cstdio>
#include<string.h>
#include<algorithm>
#include<iostream>
using namespace std;
const int M=2e5+100;
int f[M];
int d[M];
int siz[M];
int rk[M];//保存树链剖分后，节点编号对应树上位置
int id[M];//节点剖分后所对应的编号，id[rk[x]]]=x;
int top[M];
int son[M];
int n,m,s;
int tr[M*4],cnt,e,tr2[M*4];
int to[M],cost[M],head[M],nxt[M],lazy[M*4];
int a[M];
//vector<int>v[M];
struct nod
{
    int x, y;
    int c;
} node[M];
void add(int u,int v,int w)
{
    e++;
    to[e]=v;
    cost[e]=w;
    nxt[e]=head[u];
    head[u]=e;
}
void dfs1(int rt,int now,int dep)
{
    siz[now]=1;
    f[now]=rt;
    d[now]=dep;
    for(int i=head[now]; i!=-1; i=nxt[i])
    {
        int v=to[i];
        if(v==rt)continue;
        dfs1(now,v,dep+1);
        siz[now]+=siz[v];
        if(siz[v]>siz[son[now]])
            son[now]=v;
    }
}
void dfs2(int now,int t)
{
    top[now]=t;
    id[now]=++cnt;
    rk[cnt]=now;
    if(!son[now])
    {
        return;
    }
    dfs2(son[now],t);//将重链链接在一起
    for(int i=head[now]; i!=-1; i=nxt[i])
    {
        int v=to[i];
        if(v!=son[now]&&v!=f[now])
        {
            dfs2(v,v);
        }
    }

}
void pushup(int i)
{
    tr[i]=max(tr[i*2],tr[i*2+1]);
    tr2[i]=min(tr2[i*2],tr2[i*2+1]);
}
void pushdown(int i)
{
    if(lazy[i]==1)
    {
        lazy[i*2]^=1;
        lazy[i*2+1]^=1;
        lazy[i]^=1;
        tr[i*2]=-1*tr[i*2];
        tr[i*2+1]=-1*tr[i*2+1];
        tr2[i*2]=-1*tr2[i*2];
        tr2[i*2+1]=-1*tr2[i*2+1];
        swap(tr[i*2],tr2[i*2]);
        swap(tr[i*2+1],tr2[i*2+1]);
    }
}
void build(int i,int l,int r)
{
    lazy[i]=0;
    if(l==r)
    {
        tr[i]=tr2[i]=a[l];
        return ;
    }
    int mid=(l+r)/2;
    build(i*2,l,mid);
    build(i*2+1,mid+1,r);
    pushup(i);
}
void update(int i,int l,int r,int x,int y)
{
    if(l>=x&&r<=y)
    {
        tr[i]=-1*tr[i];
        tr2[i]=-1*tr2[i];
        swap(tr[i],tr2[i]);
        lazy[i]^=1;
        return;
    }
    pushdown(i);
    int mid=(l+r)/2;
    if(x<=mid)
    {
        update(i*2,l,mid,x,y);
    }
    if(y>mid)
    {
        update(i*2+1,mid+1,r,x,y);
    }
    pushup(i);
}
void update1(int i,int l,int r,int x,int t)
{
    if(l==r)
    {
        tr[i]=t;
        tr2[i]=t;
        return;
    }
    pushdown(i);
    int mid=(l+r)/2;
    if(x<=mid)
    {
        update1(i*2,l,mid,x,t);
    }
    else
    {
        update1(i*2+1,mid+1,r,x,t);
    }
    pushup(i);
}
int query(int i,int l,int r,int x,int y)
{
    if(x<=l&&y>=r)
    {
        return tr[i];
    }
    pushdown(i);
    int ans=-99999999;
    int mid=(l+r)/2;
    if(x<=mid)
        ans=max(ans,query(i*2,l,mid,x,y));
    if(y>mid)
        ans=max(ans,query(i*2+1,mid+1,r,x,y));
    return ans;
}
void qupdate(int x,int y)
{
    int fx=top[x],fy=top[y];
    while(fx!=fy)
    {
        if(d[fx]<d[fy])swap(x,y),swap(fx,fy);
        update(1,2,n,id[fx],id[x]);
        x=f[top[x]];
        fx=top[x];
    }
    if(x==y)
        return ;
    if(d[x]>d[y])
        swap(x,y);
    update(1,2,n,id[son[x]],id[y]);
}
int ask(int x,int y)
{
    int ans=-99999999;
    int fx=top[x],fy=top[y];
    while(fx!=fy)
    {
        if(d[fx]<d[fy])swap(x,y),swap(fx,fy);
        ans=max(ans,query(1,2,n,id[fx],id[x]));
        x=f[top[x]];
        fx=top[x];
    }
    if(x==y)
        return ans;
    if(d[x]>d[y])
        swap(x,y);
    ans=max(ans,query(1,2,n,id[son[x]],id[y]));
    return ans;
}

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
       scanf("%d",&n);
        cnt=0;
        e=0;
        memset(head,-1,sizeof(head));
        memset(son,0,sizeof(son));
        for(int i=1; i<n; i++)
        {
            int x,y,t;
            scanf("%d%d%d",&x,&y,&t);
            add(x,y,t);
            add(y,x,t);
            node[i].x=x;
            node[i].y=y;
            node[i].c=t;
        }
        dfs1(0,1,1);
        dfs2(1,1);
        for(int i=1; i<n; i++)
        {
            int x=node[i].x;
            int y=node[i].y;
            if(d[x]<d[y])
                swap(x,y);
            a[id[x]]=node[i].c;
        }
         build(1,2,n);
        char op[20];
        int x,y;
        scanf("%s",op);
        while(op[0]!='D')
        {
            scanf("%d%d",&x,&y);
            if(op[0]=='Q')
            {
                printf("%d\n",ask(x,y));
            }
            else if(op[0]=='C')
            {
                int u=node[x].x;
                int v=node[x].y;
                if(d[u]<d[v])
                    swap(u,v);
                update1(1,2,n,id[u],y);
            }
            else
            {
                qupdate(x,y);
            }
            scanf("%s",op);
        }
    }

    return 0;
}