POJ3237:Tree

题目链接

树链剖分套线段树。很经典的题。
单点修改，区间最大值，区间取反（乘-1）。

再写还是没一遍A。。很羞耻。
（凭什么G++编译比C++快一倍还多。。）

#include <cstdio>
#include <iostream>
#include <queue>
#include <vector>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <stack>
#define N 10005 
#define M 200005
#define INF 1e9
#define mod 1000000
using namespace std;
typedef long long ll;
typedef pair<ll,ll> pa;

int read()
{
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
    while(isdigit(ch)){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return x*f;
}

int n,m,cnt,ind,T;
int b[N<<1],p[N],nextedge[N<<1],w[N<<1];
int deep[N],fa[N],son[N],sz[N],pos[N],ftree[N],top[N],a[N];

class Seg_Tree{
    public:
        int l,r,mx,tag,mn;
}e[N<<2];

void Add(int x,int y,int z)
{
    cnt++;
    b[cnt]=y;
    nextedge[cnt]=p[x];
    p[x]=cnt;
    w[cnt]=z;
}

void Anode(int x,int y,int z){
    Add(x,y,z);Add(y,x,z);
}

void Dfs(int x)
{
    sz[x]=1;
    for(int i=p[x];i;i=nextedge[i])
    {
        int v=b[i];
        if(v==fa[x]) continue;
        deep[v]=deep[x]+1;fa[v]=x;
        Dfs(v);
        sz[x]+=sz[v];
        son[x]=sz[son[x]]>sz[v]?son[x]:v;
    }
}

void dfs(int x,int root)
{
    pos[x]=++ind;ftree[pos[x]]=x;
    top[x]=root;
    if(!son[x]) return;
    dfs(son[x],root);
    for(int i=p[x];i;i=nextedge[i])
    {
        int v=b[i];
        if(v==fa[x]||v==son[x]) continue;
        dfs(v,v);
    }
}

void Input_Init()
{
    n=read();
    cnt=ind=0;
    for(int i=1;i<=n;i++) pos[i]=ftree[i]=p[i]=son[i]=sz[i]=deep[i]=fa[i]=0;
    for(int i=1;i<n;i++)
    {
        static int x,y,z;
        x=read(),y=read(),z=read();
        Anode(x,y,z);
    }
}

void Edge_To_Node()
{
    for(int i=1;i<cnt;i+=2)
    {
        int x=b[i],y=b[i+1];
        if(deep[x]<deep[y]) swap(x,y);
        a[x]=w[i];
    }
}

void pushup(int p)
{
    e[p].mn=min(e[p<<1].mn,e[p<<1|1].mn);
    e[p].mx=max(e[p<<1].mx,e[p<<1|1].mx);
}

void Swap(int p){
    int tmp=e[p].mx;e[p].mx=-e[p].mn;e[p].mn=-tmp;
}

void pushdown(int p)
{
    if(e[p].tag)
    {
        e[p].tag=0;e[p<<1].tag^=1;e[p<<1|1].tag^=1;
        Swap(p<<1);Swap(p<<1|1);
        return;
    }   
}

void Build(int p,int l,int r)
{
    e[p].l=l,e[p].r=r;e[p].tag=0;
    if(l==r)
    {
        e[p].mn=e[p].mx=a[ftree[l]];
        return;
    }
    int mid=l+r>>1;
    Build(p<<1,l,mid);Build(p<<1|1,mid+1,r);
    pushup(p);
}

int Query_Max(int p,int x,int y)
{
    int l=e[p].l,r=e[p].r,mid=l+r>>1;
    if(l==x&&y==r)
        return e[p].mx;
    pushdown(p);
    if(y<=mid) return Query_Max(p<<1,x,y);
    if(x>mid) return Query_Max(p<<1|1,x,y);
    return max(Query_Max(p<<1,x,mid),Query_Max(p<<1|1,mid+1,y));
}

int Query(int x,int y)
{
    int fx=top[x],fy=top[y];
    int rtn=-INF;
    while(fx!=fy)
    {
        if(deep[fx]<deep[fy]) swap(x,y),swap(fx,fy);
        rtn=max(rtn,Query_Max(1,pos[fx],pos[x]));
        x=fa[fx];fx=top[x];
    }
    if(deep[x]<deep[y]) swap(x,y);
    if(x!=y) rtn=max(rtn,Query_Max(1,pos[y]+1,pos[x]));
    return rtn;
}

void Solve_Change(int p,int x,int y)
{
    int l=e[p].l,r=e[p].r,mid=l+r>>1;
    if(l==r&&x==r)
    {
        e[p].mn=e[p].mx=y;
        e[p].tag=0;return;
    }
    pushdown(p);
    if(x<=mid) Solve_Change(p<<1,x,y);
    else Solve_Change(p<<1|1,x,y);
    pushup(p); 
}

void Change(int x,int y)
{
    int xx=b[x<<1],yy=b[(x<<1)-1];
    if(deep[xx]<deep[yy]) swap(xx,yy);
    Solve_Change(1,pos[xx],y);
}

void Solve_Negate(int p,int x,int y)
{
    int l=e[p].l,r=e[p].r,mid=l+r>>1;
    if(l==x&&y==r)
    {
        Swap(p);e[p].tag^=1;
        return;
    }
    pushdown(p);
    if(y<=mid) Solve_Negate(p<<1,x,y);
    else if(x>mid) Solve_Negate(p<<1|1,x,y);
    else Solve_Negate(p<<1,x,mid),Solve_Negate(p<<1|1,mid+1,y);
    pushup(p);
}

void Negate(int x,int y)
{
    int fx=top[x],fy=top[y];
    while(fx!=fy)
    {
        if(deep[fx]<deep[fy]) swap(x,y),swap(fx,fy);
        Solve_Negate(1,pos[fx],pos[x]);
        x=fa[fx],fx=top[x];
    }
    if(x==y) return;
    if(deep[x]<deep[y]) swap(x,y);
    Solve_Negate(1,pos[y]+1,pos[x]);
}

void Solve()
{
    char ch[10];
    while(true)
    {
        scanf("%s",ch);
        if(ch[0]=='D') break;
        static int x,y;
        x=read(),y=read();
        if(ch[0]=='Q') printf("%d\n",Query(x,y));
        else if(ch[0]=='C') Change(x,y);
        else Negate(x,y);
    }
}

int main()
{
    T=read();
    while(T--)
    {
        Input_Init();
        Dfs(1);dfs(1,1);
        Edge_To_Node();Build(1,1,n);
        Solve();
    }
    return 0;
}