BZOJ3052: [wc2013]糖果公园

学了一发带修改树上莫队，然后就发现这道题很简单了。
带修改的莫队：由原来的二维状态$(l,r)$改为三维$(l,r,x)$，第三维度$x$表示的是当前这个状态经过了$x$次修改，按$l$所在的块为第一关键字，$r$所在的块为第二关键字，$x$为第三关键字排序，设$K$为块的数量，将$l,r$所在块相同的情况视作同一组
那么对于同一组的情况，$x$是单调递增的，$x$的移动是$O(n)$的，共有$K^2$种情况，所以是$O(nK^2)$的
对于同一组$l,r$的移动是$O(n/K)$的，共有$n$种，所以是$O(n^2/K)$的
组与组之间的转移，每次$O(n)$，共$K^2$组，所以是$O(nK^2)$的
然后总复杂度就是$O(nK^2+n^2/K)$，当$K$取$n^{1/3}$时，就有最优复杂度$O(n^{5/3})$
然后树上莫队：用括号序列，第一次访问$'('$，退出这个点时再加一个$')'$，记录每个点$'(',')'$的位置，那么考虑树上两点之间的问题，如果一点是另一点祖先，那$(first[x],first[y])$区间内除了链上的点，其他点都出现了偶数次，如果LCA不在这两个点中，那么$(last[x],first[y])$区间内也是除了链上的点，其余点都出现偶数次，但是两点的LCA不在区间内，所以计算答案的时候要算进去

然后把这两个合起来，这道题就变成模板题啦
TLE了一发好像卡了很久的评测QAQ

---

---

code：

#include<set>
#include<map>
#include<deque>
#include<queue>
#include<stack>
#include<cmath>
#include<ctime>
#include<bitset>
#include<string>
#include<vector>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<climits>
#include<complex>
#include<iostream>
#include<algorithm>
#define ll long long
using namespace std;

const int maxn = 210000;
const int maxl = 20;
struct edge
{
    int y,nex;
    edge(){}
    edge(int _y,int _nex){y=_y;nex=_nex;}
}a[maxn<<1]; int len,fir[maxn];
void ins(int x,int y){a[++len]=edge(y,fir[x]);fir[x]=len;}
int n,M,Q,v[maxn]; ll w[maxn],Cw[maxn];
int vi[maxn],upd[maxn][2];
int fa[maxn][maxl],dep[maxn];
int s[maxn],fi[maxn],la[maxn],cnt;
int id[maxn],L[maxn],N;
void dfs(int x)
{
    s[++cnt]=x; fi[x]=cnt;
    for(int k=fir[x];k;k=a[k].nex)
    {
        int y=a[k].y;
        if(y!=fa[x][0])
        {
            fa[y][0]=x;
            dep[y]=dep[x]+1;
            dfs(y);
        }
    }
    s[++cnt]=x; la[x]=cnt;
}
int LCA(int x,int y)
{
    if(dep[x]<dep[y]) swap(x,y);
    for(int i=maxl-1;i>=0;i--)if(dep[x]-(1<<i)>=dep[y])
        x=fa[x][i];
    if(x==y)return x;
    for(int i=maxl-1;i>=0;i--)
        if(fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
struct qu
{
    int l,r,x,k,i;
    qu(){}
    qu(int _l,int _r,int _x,int _k,int _i){l=_l;r=_r;x=_x;k=_k;i=_i;}
}q[maxn]; ll ans[maxn]; int m;
bool cmp(qu x,qu y)
{
    return id[x.l]==id[y.l]?(id[x.r]==id[y.r]?x.x<y.x:id[x.r]<id[y.r]):id[x.l]<id[y.l];
}
int num[maxn],ti[maxn];
void pre()
{
    n=cnt; N=pow(n,2.0/3);
    for(int i=1;i<=n;i++) id[i]=(i-1)/N+1;
    for(int i=1;i<=id[n];i++) L[i]=(i-1)*N+1;
    L[id[n]+1]=n+1;
}
void solve()
{
    pre();
    sort(q+1,q+m+1,cmp);
    for(int i=1;i<=m;)
    {
        int t1=id[q[i].l],t2=id[q[i].r];
        int j;
        for(j=i+1;j<=m&&id[q[j].l]==t1&&id[q[j].r]==t2;j++);
        j--;
        for(int l=1;l<=n;l++) vi[l]=v[l],ti[l]=0;
        for(int l=1;l<=M;l++) num[l]=0;
        int nl=L[t1],nr=nl-1,nx=0; ll ns=0;
        for(;i<=j;i++)
        {
            while(nx<q[i].x) 
            {
                nx++; int up1=upd[nx][0],up2=upd[nx][1];
                if(ti[up1]==1)
                {
                    ns-=w[num[vi[up1]]]*Cw[vi[up1]],num[vi[up1]]--;
                    num[up2]++,ns+=w[num[up2]]*Cw[up2];
                }
                vi[up1]=up2;
            }
            while(nr<q[i].r)
            {
                nr++;
                if(ti[s[nr]]==1) ns-=w[num[vi[s[nr]]]]*Cw[vi[s[nr]]],ti[s[nr]]=2,num[vi[s[nr]]]--;
                else ti[s[nr]]=1,num[vi[s[nr]]]++,ns+=w[num[vi[s[nr]]]]*Cw[vi[s[nr]]];
            }
            while(nl>q[i].l)
            {
                nl--;
                if(ti[s[nl]]==1) ns-=w[num[vi[s[nl]]]]*Cw[vi[s[nl]]],ti[s[nl]]=2,num[vi[s[nl]]]--;
                else ti[s[nl]]=1,num[vi[s[nl]]]++,ns+=w[num[vi[s[nl]]]]*Cw[vi[s[nl]]];
            }
            while(nr>q[i].r)
            {
                if(ti[s[nr]]==1) ns-=w[num[vi[s[nr]]]]*Cw[vi[s[nr]]],ti[s[nr]]=0,num[vi[s[nr]]]--;
                else ti[s[nr]]=1,num[vi[s[nr]]]++,ns+=w[num[vi[s[nr]]]]*Cw[vi[s[nr]]];
                nr--;
            }
            while(nl<q[i].l)
            {
                if(ti[s[nl]]==1) ns-=w[num[vi[s[nl]]]]*Cw[vi[s[nl]]],ti[s[nl]]=0,num[vi[s[nl]]]--;
                else ti[s[nl]]=1,num[vi[s[nl]]]++,ns+=w[num[vi[s[nl]]]]*Cw[vi[s[nl]]];
                nl++;
            }
            ans[q[i].i]=ns;
            if(q[i].k!=-1) 
                ans[q[i].i] += w[num[vi[q[i].k]]+1]*Cw[vi[q[i].k]];
        }
    } 
}

int main()
{
    memset(fir,0,sizeof fir); len=0;

    scanf("%d%d%d",&n,&M,&Q);
    for(int i=1;i<=M;i++) scanf("%lld",&Cw[i]);
    for(int i=1;i<=n;i++) scanf("%lld",&w[i]);
    for(int i=1;i<n;i++)
    {
        int x,y; scanf("%d%d",&x,&y);
        ins(x,y); ins(y,x);
    }
    for(int i=1;i<=n;i++) scanf("%d",&v[i]);
    cnt=0; dfs(1);
    for(int i=1;i<maxl;i++)
        for(int x=1;x<=n;x++)if(dep[x]>=1<<i)
        {
            fa[x][i]=fa[fa[x][i-1]][i-1];
        }
    int tt=0; m=0;
    for(int i=1;i<=Q;i++)
    {
        int x,y,c;
        scanf("%d%d%d",&c,&x,&y);
        if(c==0){ tt++; upd[tt][0]=x; upd[tt][1]=y; }
        else
        {
            m++;
            if(dep[x]>dep[y]) swap(x,y);
            int D=LCA(x,y);
            if(D==x) q[m]=qu(fi[x],fi[y],tt,-1,m);
            else
            {
                if(la[x]<fi[y])q[m]=qu(la[x],fi[y],tt,D,m);
                else q[m]=qu(la[y],fi[x],tt,D,m);
            }
        }
    }
    solve();
    for(int i=1;i<=m;i++) printf("%lld\n",ans[i]);

    return 0;
}