【GDKOI2015】【JZOJ 4005】树 括号序

Description

Analysis

KD啊OJ上交不了这道题，我只好co标自己打拍验证程序了(面壁)
这题是GDKOI2015Day1最后一题
看数据范围，N<=50000，log^2好像能过，链剖？
因为只有10种颜色，所以开10棵线段树维护每种颜色链上的值
第一问子树内好恶心，线段树怎么维护
等一等，子树？想一想有什么东西可以把子树转化成连续一段序列？

17.7.11 update：下面写的别看了QAQ，别人写的更好dfs序的常见用法整理

没错，dfs序
但是dfs序有两种常见套路：

一：记录dfs序（其实没卵用不想看的看下面），即到一个新的点就记录该点编号。线段树记录的是该点到根的权值和。那么修改一个点的权值会对子树内所有点造成影响，所以线段树里要修改区间$[dfn(x),dfn(x)+size(x)-1]$一整段。size(x)表示x子树内点数。查询u到v路径，可以转化为单点查询u,v,lca(u,v)。所以这种方法支持树上单点修改，链查询。

二：记录括号序，即在到达和离开该点的时候各记录该点编号，那么序列长度是2N的。x的子树在序列里对应区间$[l(x),r(x)]$，l[x]表示第一次出现x的位置，r[x]表示最后出现x的位置。线段树记录的是单点的权值。在l[x]位置赋为权值，r[x]赋为权值的相反数。做前缀和，那么x到根的权值和就是$sum[l(x)]$。这个前缀和用树状数组最简单。那么修改单点权值的话线段树单点修改即可；修改一个点的整个子树，也只需要提取出子树区间，区间修改。支持单点修改，子树修改。链查询，子树查询。

两种套路都不能支持链上的修改，且维护的信息需满足可减性，而如果要维护最值就GG了
出题人甚至可以把树动起来，然后就变成ETT了
好像第二种更高级些，适用范围更广
这题也只能用第二种，但是有交换颜色，所以要打线段树换点
至于换点，可以一次搞定（参见代码实现），或者可以用一个区间懒标记
而且时间复杂度仅为$O((n+m)logn)$

Code

#include<cstdio>
#include<cmath>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,b,a) for(int i=b;i>=a;i--)
#define efo(i,v) for(int i=last[v];i;i=next[i])
using namespace std;
const int N=50005,M=N*2,V=4000005;
int n,tot,now,co[N],b[N],to[M],next[M],last[N];
int m,root[N],c[M],dep[N],l[N],r[N],f[N][20];
struct segment
{
    int ls,rs,x;
}a[V];
void link(int u,int v)
{
    to[++tot]=v,next[tot]=last[u],last[u]=tot;
}
void dfs(int v,int fr,int d)
{
    c[++m]=v,f[v][0]=fr,dep[v]=d,l[v]=m;
    efo(i,v)
    {
        int u=to[i];
        if(u==fr) continue;
        dfs(u,v,d+1);
    }
    c[++m]=v,r[v]=m;
}
int getlca(int u,int v)
{
    if(dep[u]<dep[v]) swap(u,v);
    fd(i,log2(dep[u]),0)
        if(dep[f[u][i]]>=dep[v]) u=f[u][i];
    if(u==v) return u;
    fd(i,log2(dep[u]),0)
        if(f[u][i]!=f[v][i]) u=f[u][i],v=f[v][i];
    return f[u][0];
}
void up(int v)
{
    a[v].x=a[a[v].ls].x+a[a[v].rs].x;
}
void change(int &v,int l,int r,int x,int z)
{
    if(!v) v=++tot;
    if(l==r)
    {
        a[v].x=z;
        return;
    }
    int mid=(l+r)>>1;
    if(x<=mid) change(a[v].ls,l,mid,x,z);
    else change(a[v].rs,mid+1,r,x,z);
    up(v);
}
void exchange(int &u,int &v,int l,int r,int x,int y)
{
    if(!u) u=++tot;
    if(!v) v=++tot;
    if(l==x && r==y)
    {
        swap(u,v);
        return;
    }
    int mid=(l+r)>>1;
    if(y<=mid) exchange(a[u].ls,a[v].ls,l,mid,x,y);
    else
    if(x>mid) exchange(a[u].rs,a[v].rs,mid+1,r,x,y);
    else
    {
        exchange(a[u].ls,a[v].ls,l,mid,x,mid);
        exchange(a[u].rs,a[v].rs,mid+1,r,mid+1,y);
    }
    up(u);up(v);
}
int query(int v,int l,int r,int x,int y)
{
    if(!v) return 0;
    if(l==x && r==y) return a[v].x;
    int mid=(l+r)>>1;
    if(y<=mid) return query(a[v].ls,l,mid,x,y);
    else
    if(x>mid) return query(a[v].rs,mid+1,r,x,y);
    else
    return query(a[v].ls,l,mid,x,mid)+query(a[v].rs,mid+1,r,mid+1,y);
}
int main()
{
    freopen("4005.in","r",stdin);
    freopen("4005.out","w",stdout);
    int _,x,y,z,lca;
    char ch;
    scanf("%d",&n);
    fo(i,1,n) scanf("%d",&co[i]);
    fo(i,1,n) scanf("%d",&b[i]);
    fo(i,1,n-1)
    {
        scanf("%d %d",&x,&y);x++,y++;
        link(x,y),link(y,x);
    }
    dfs(1,0,1);
    fo(i,1,n)
    {
        change(root[co[i]],1,m,l[i],b[i]);
        change(root[co[i]],1,m,r[i],-b[i]);
    }
    fo(j,1,log2(n))
        fo(i,1,n) f[i][j]=f[f[i][j-1]][j-1];
    scanf("%d\n",&_);
    while(_--)
    {
        scanf("%c",&ch);
        if(ch=='A')
        {
            scanf("sk %d %d %d\n",&x,&y,&z);x++,y++;
            lca=getlca(x,y);
            int ans=query(root[z],1,m,1,l[x])+query(root[z],1,m,1,l[y])-
                    query(root[z],1,m,1,l[lca])-query(root[z],1,m,1,l[f[lca][0]]);
            printf("%d\n",ans);
        }
        else
        if(ch=='C')
        {
            scanf("hange %d %d %d\n",&x,&y,&z);x++;
            exchange(root[y],root[z],1,m,l[x],r[x]);
        }
        else
        {
            scanf("et %d %d %d\n",&x,&y,&z);x++;
            int col=-1;
            fo(i,0,9)
                if(query(root[i],1,m,l[x],l[x]))
                {
                    change(root[i],1,m,l[x],0);
                    change(root[i],1,m,r[x],0);
                    break;
                }
            change(root[y],1,m,l[x],z);
            change(root[y],1,m,r[x],-z);
        }
    }
    return 0;
}