BZOJ 3052: [wc2013]糖果公园

蒟蒻去做糖果公园了QAQ
讲道理速度能排rk11很开心啊~

其实前人之述备矣 复杂度分析 推荐一波栋爷爷的题解

带修改莫队其实就是多一维时间 所以你需要资瓷修改和撤销修改
然后就跟没修改的莫队其实差不多了

关于树上莫队 很明显就是用dfs序这个套路
发现出现次数为奇数次的才会算到 栋老师题解那里也有说到
就讨论一下 lca是不是那两个点就好
（代码参考 抄袭 了纪中栋栋 不过我觉得我写的挺短挺好看的）

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N=1e5+5;
inline int read(){
    int x=0,f=1; char ch=getchar();
    while(ch<'0' || ch>'9'){if(ch=='-')f=-1; ch=getchar();}
    while(ch>='0' && ch<='9'){x=(x<<1)+(x<<3)+ch-'0'; ch=getchar();}
    return x*f;
}
int nex[N<<1],go[N<<1],fir[N],len,f[N][17];
int L[N],R[N],tot,dep[N],a[N<<1],bl[N<<1];
void ins(int x,int y){
    nex[++len]=fir[x],fir[x]=len,go[len]=y;
}
void dfs(int x,int fa){
    f[x][0]=fa,dep[x]=dep[fa]+1;
    L[x]=++tot,a[tot]=x;
    for(int i=1;i<17;++i)
        if(f[x][i-1])f[x][i]=f[f[x][i-1]][i-1];
        else break;
    for(int k=fir[x];k;k=nex[k]){
        int y=go[k]; if(y==fa)continue;
        dfs(y,x);
    }
    R[x]=++tot,a[tot]=x;
}
int lca(int x,int y){
    if(dep[x]<dep[y])swap(x,y);
    int i;
    if(dep[x]!=dep[y]){
        i=int(log(dep[x]-dep[y])/log(2));
        for(;~i;--i) if(dep[f[x][i]]>dep[y])x=f[x][i];
        x=f[x][0];
    }
    if(x==y)return x;
    i=int(log(dep[x])/log(2));
    for(;~i;--i) if(f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
    return f[x][0];
}
struct ask{
    int l,r,x,id,w; bool p;
}o[N]; int ol;
int Cmp(ask x,ask y){
    if(bl[x.l]!=bl[y.l])return bl[x.l]<bl[y.l];
    if(bl[x.r]!=bl[y.r])return bl[x.r]<bl[y.r];
    return x.x<y.x;
}
int n,m,q,v[N],w[N],c[N],d[N][3],dl,cnt[N];
LL s,ans[N]; bool e[N];
void change(int x){
    e[x]^=1;
    if(e[x])
        s+=1ll*w[++cnt[c[x]]]*v[c[x]];
    else
        s-=1ll*w[cnt[c[x]]--]*v[c[x]];
}
void ins(int x){
    if(e[d[x][2]]){
        s-=1ll*w[cnt[d[x][0]]--]*v[d[x][0]];
        s+=1ll*w[++cnt[d[x][1]]]*v[d[x][1]];
    }
    c[d[x][2]]=d[x][1];
}
void del(int x){
    if(e[d[x][2]]){
        s-=1ll*w[cnt[d[x][1]]--]*v[d[x][1]];
        s+=1ll*w[++cnt[d[x][0]]]*v[d[x][0]];
    }
    c[d[x][2]]=d[x][0];
}
int main(){
    int i,l,r; n=read(),m=read(),q=read();
    for(i=1;i<=m;++i)v[i]=read();
    for(i=1;i<=n;++i)w[i]=read();
    for(i=1;i<n;++i){
        int x=read(),y=read();
        ins(x,y),ins(y,x);
    }
    dfs(1,0);
    for(i=1;i<=n;++i)c[i]=read();
    for(i=1;i<=q;++i){
        int t=read();
        if(!t){
            int x=read(),y=read();
            d[++dl][0]=c[x],
            d[dl][1]=c[x]=y,d[dl][2]=x;
        }
        else{
            int x=read(),y=read();
            t=lca(x,y);
            if(t==x||t==y){
                o[++ol].p=0;
                l=L[x],r=L[y];
                if(l>r)l^=r^=l^=r;
            }
            else{
                o[++ol].p=1,o[ol].w=t;
                if(L[x]>L[y])swap(x,y);
                l=R[x],r=L[y];
            }
            o[ol].l=l,o[ol].r=r;
            o[ol].id=ol,o[ol].x=dl;
        }
    }
    for(i=dl;i;--i) c[d[i][2]]=d[i][0];
    int bk=2100,x=0;
    for(i=1;i<=n*2;++i) bl[i]=(i-1)/bk+1;
    sort(o+1,o+1+ol,Cmp);
    l=r=1; change(a[1]);
    for(i=1;i<=ol;++i){
        while(x>o[i].x) del(x),--x;
        while(x<o[i].x) ++x,ins(x);
        while(l>o[i].l) --l,change(a[l]);
        while(l<o[i].l) change(a[l]),++l;
        while(r>o[i].r) change(a[r]),--r;
        while(r<o[i].r) ++r,change(a[r]);
        if(o[i].p) change(o[i].w);
        ans[o[i].id]=s;
        if(o[i].p) change(o[i].w);
    }
    for(i=1;i<=ol;++i) printf("%lld\n",ans[i]);
    return 0;
}