bzoj3531: [Sdoi2014]旅行

用类似主席树的方法动态开点链剖+线段树，分宗教维护线段树。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int read()
{
    char ch=getchar();int f=0;
    while(ch<'0'||ch>'9')ch=getchar();
    while(ch>='0'&&ch<='9'){f=(f<<1)+(f<<3)+ch-'0';ch=getchar();}
    return f;
}
int n,m,tot,cnt,size[100005],dep[100005],root[100005],mx[6000005],sum[6000005],top[100005];
int ch[6000005][2],a[100005],b[100005],head[100005],pos[100005],npos[100005],fa[100005],son[100005];
struct node
{
    int from;
    int to;
    int next;
}edge[200005];
void add(int u,int v)
{
    edge[tot].from=u;
    edge[tot].to=v;
    edge[tot].next=head[u];
    head[u]=tot++;
}
void dfs1(int x)
{
    size[x]=1;
    dep[x]=dep[fa[x]]+1;
    for(int i=head[x];i!=-1;i=edge[i].next)
    {
        if(edge[i].to!=fa[x])
        {
            fa[edge[i].to]=x;
            dfs1(edge[i].to);
            if(size[son[x]]<size[edge[i].to])
            son[x]=edge[i].to;
            size[x]+=size[edge[i].to];
        }
    }
}
void dfs2(int x)
{
    //cout<<x<<" ";
    pos[x]=++cnt;
    npos[cnt]=x;
    if(son[fa[x]]==x) top[x]=top[fa[x]];
    else top[x]=x;
    if(son[x])
    dfs2(son[x]);
    for(int i=head[x];i!=-1;i=edge[i].next)
    {
        if(edge[i].to!=fa[x]&&edge[i].to!=son[x])
        dfs2(edge[i].to);
    }
}
void modify(int &now,int l,int r,int x,int w)
{
    if(!now) now=++cnt;
    if(l==r)
    {
        mx[now]=sum[now]=w;
        return;
    }
    int mid=l+r>>1;
    if(x<=mid) modify(ch[now][0],l,mid,x,w);
    else modify(ch[now][1],mid+1,r,x,w);
    mx[now]=max(mx[ch[now][0]],mx[ch[now][1]]);
    sum[now]=sum[ch[now][0]]+sum[ch[now][1]];
}
void remove(int now,int l,int r,int x)
{
    if(l==r)
    {
        mx[now]=sum[now]=0;
        return;
    }
    int mid=l+r>>1;
    if(x<=mid) remove(ch[now][0],l,mid,x);
    else remove(ch[now][1],mid+1,r,x);
    mx[now]=max(mx[ch[now][0]],mx[ch[now][1]]);
    sum[now]=sum[ch[now][0]]+sum[ch[now][1]];
}
int querys(int p,int l,int r,int x,int y)
{
    int mid=l+r>>1;
    if(l==x&&r==y)
    return sum[p];
    if(y<=mid)
    return querys(ch[p][0],l,mid,x,y);
    else if(x>mid)
    return querys(ch[p][1],mid+1,r,x,y);
    else return querys(ch[p][0],l,mid,x,mid)+querys(ch[p][1],mid+1,r,mid+1,y);
}
int qmax(int p,int l,int r,int x,int y)
{
    int mid=l+r>>1;
    if(l==x&&r==y)
    return mx[p];
    if(y<=mid)
    return qmax(ch[p][0],l,mid,x,y);
    if(x>mid) return qmax(ch[p][1],mid+1,r,x,y);
    return max(qmax(ch[p][0],l,mid,x,mid),qmax(ch[p][1],mid+1,r,mid+1,y));
}
int querys(int x,int y)
{
    int rt=root[b[x]],fx=top[x],fy=top[y],ans=0;
    while(fx!=fy)
    {
        if(dep[fx]<dep[fy])
        {
            swap(x,y);
            swap(fx,fy);
        }
        ans+=querys(rt,1,n,pos[fx],pos[x]);
        x=fa[fx],fx=top[x];
    }
    if(dep[x]<dep[y])
    swap(x,y);
    //cout<<rt<<" "<<pos[y]<<" "<<pos[x];
    ans+=querys(rt,1,n,pos[y],pos[x]);
    return ans;
}
int qmax(int x,int y)
{
    int rt=root[b[x]],fx=top[x],fy=top[y],ans=0;
    while(fx!=fy)
    {
        if(dep[fx]<dep[fy])
        {
            swap(x,y);
            swap(fx,fy);
        }
        ans=max(ans,qmax(rt,1,n,pos[fx],pos[x]));
        x=fa[fx],fx=top[x];
    }
    if(dep[x]<dep[y])
    swap(x,y);
    //cout<<pos[y]<<" "<<pos[x]<<endl;
    ans=max(ans,qmax(rt,1,n,pos[y],pos[x]));
    return ans;
}
int main()
{
    n=read(),m=read();
    memset(head,-1,sizeof(head));
    for(int i=1;i<=n;i++)
    {
        a[i]=read(),b[i]=read();
    }
    for(int i=1;i<n;i++)
    {
        int u=read(),v=read();
        add(u,v);
        add(v,u);
    }
    dfs1(1);
    dfs2(1);
    cnt=0;
    //for(int i=1;i<=n;i++)
    //cout<<pos[i]<<" ";
    for(int i=1;i<=n;i++)
    {
        modify(root[b[i]],1,n,pos[i],a[i]);
    }
    //cout<<"ok";
    char s[15];
    for(int i=1;i<=m;i++)
    {
        int x,y;
        scanf("%s%d%d",s,&x,&y);
        if(s[1]=='C')
        {
            remove(root[b[x]],1,n,pos[x]);
            modify(root[y],1,n,pos[x],a[x]);
            //cout<<a[i];
            b[x]=y;
        }
        else if(s[1]=='W')
        {
            modify(root[b[x]],1,n,pos[x],y);
            a[x]=y;
        }
        else if(s[1]=='S') printf("%d\n",querys(x,y));
        else printf("%d\n",qmax(x,y));
    }
}