bzoj3531: [Sdoi2014]旅行

动态开点什么的真的很迷。。。

这猥琐样感觉就是没merge的线段树合并。。

练练树剖咯

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;

struct node
{
    int x,y,next;
}a[210000];int len,last[110000];
void ins(int x,int y)
{
    len++;
    a[len].x=x;a[len].y=y;
    a[len].next=last[x];last[x]=len;
}
int fa[110000],dep[110000],son[110000],tot[110000];
void pre_tree_node(int x)
{
    tot[x]=1;son[x]=0;
    for(int k=last[x];k;k=a[k].next)
    {
        int y=a[k].y;
        if(y!=fa[x])
        {
            fa[y]=x;
            dep[y]=dep[x]+1;
            pre_tree_node(y);
            if(tot[son[x]]<tot[y])son[x]=y;
            tot[x]+=tot[y];
        }
    }
}
int z,ys[110000],top[110000];
void pre_tree_edge(int x,int tp)
{
    ys[x]=++z;top[x]=tp;
    if(son[x]!=0)pre_tree_edge(son[x],tp);
    for(int k=last[x];k;k=a[k].next)
    {
        int y=a[k].y;
        if(y!=son[x]&&y!=fa[x])pre_tree_edge(y,y);
    }
}

//---------init-----------------------

struct trnode
{
    int lc,rc,c,d;
}tr[21000000];int trlen,rt[110000];
int maketree(int now,int l,int r,int p,int d)
{
    if(now==0)
    {
        now=++trlen;
        tr[now].lc=tr[now].rc=0;
    }
    if(l==r){tr[now].c=tr[now].d=d; return now;}
    else
    {
        int mid=(l+r)/2;
        if(p<=mid)tr[now].lc=maketree(tr[now].lc,l,mid,p,d);
        else       tr[now].rc=maketree(tr[now].rc,mid+1,r,p,d);
        tr[now].c=max(tr[tr[now].lc].c,tr[tr[now].rc].c);
        tr[now].d=tr[tr[now].lc].d+tr[tr[now].rc].d;
        return now;
    }
}
int getmax(int now,int l,int r,int ll,int rr)
{
    if(now==0)return 0;
    if(ll==l&&r==rr)return tr[now].c;
    int mid=(l+r)/2;
         if(rr<=mid)  return getmax(tr[now].lc,l,mid,ll,rr);
    else if(mid+1<=ll)return getmax(tr[now].rc,mid+1,r,ll,rr);
    else return max(getmax(tr[now].lc,l,mid,ll,mid),getmax(tr[now].rc,mid+1,r,mid+1,rr));
}
int getsum(int now,int l,int r,int ll,int rr)
{
    if(now==0)return 0;
    if(ll==l&&r==rr)return tr[now].d;
    int mid=(l+r)/2;
         if(rr<=mid)  return getsum(tr[now].lc,l,mid,ll,rr);
    else if(mid+1<=ll)return getsum(tr[now].rc,mid+1,r,ll,rr);
    else return getsum(tr[now].lc,l,mid,ll,mid)+getsum(tr[now].rc,mid+1,r,mid+1,rr);
}

//-----------------seg_tree------------

int n,bel[110000];
int solve(int x,int y)
{
    int ans=0;
    int tx=top[x],ty=top[y],wy=bel[x];
    while(tx!=ty)
    {
        if(dep[tx]>dep[ty])swap(x,y),swap(tx,ty);
        ans=max(ans,getmax(rt[wy],1,n,ys[ty],ys[y]));
        y=fa[ty],ty=top[y];
    }
    if(dep[x]>dep[y])swap(x,y);
    ans=max(ans,getmax(rt[wy],1,n,ys[x],ys[y]));
    return ans;
}
int solve2(int x,int y)
{
    int tx=top[x],ty=top[y],ans=0,wy=bel[x];
    while(tx!=ty)
    {
        if(dep[tx]>dep[ty])swap(tx,ty),swap(x,y);
        ans+=getsum(rt[wy],1,n,ys[ty],ys[y]);
        y=fa[ty],ty=top[y];
    }
    if(dep[x]>dep[y])swap(x,y);
    ans+=getsum(rt[wy],1,n,ys[x],ys[y]);
    return ans;
}
int w[110000];
char ss[10];
int main()
{
    int Q;
    scanf("%d%d",&n,&Q);
    for(int i=1;i<=n;i++)scanf("%d%d",&w[i],&bel[i]);
    for(int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        ins(x,y);ins(y,x);
    }
    dep[1]=1;fa[1]=0;pre_tree_node(1);
    z=0;pre_tree_edge(1,1);
    
    for(int i=1;i<=n;i++)rt[bel[i]]=maketree(rt[bel[i]],1,n,ys[i],w[i]);
    while(Q--)
    {
        int x,y;
        scanf("%s%d%d",ss+1,&x,&y);
        if(ss[2]=='C')
        {
            rt[bel[x]]=maketree(rt[bel[x]],1,n,ys[x],0);
            bel[x]=y;
            rt[bel[x]]=maketree(rt[bel[x]],1,n,ys[x],w[x]);
        }
        if(ss[2]=='W')
        {
            rt[bel[x]]=maketree(rt[bel[x]],1,n,ys[x],y);
            w[x]=y;
        }
        if(ss[2]=='M')printf("%d\n",solve(x,y));
        if(ss[2]=='S')printf("%d\n",solve2(x,y));
    }
    return 0;
}