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;}