bzoj 3052/洛谷 P4074/uoj 58 [WC2013]糖果公园（树上带修莫队）

题意：

UOJ地址      洛谷地址
BZOJ因为版权挂了。。。

思路：

     比较裸的一道树上带修莫队，我们只要维护一下某个糖出现了几次，然后更新答案即可

错误及反思：

     本来是来连板子和熟练度了，然而非常坑地卡常了，在我把vector换成链式前向星，所有函数都加了inline以后，终于在洛谷上过了（UOJ没有卡常，而且加了链式前向星在UOJ上反而慢了），只能说真的有点看不懂测评机的机制

代码：

#include<bits/stdc++.h>
using namespace std;
const int N = 110000;
int blocks,block[N],n,m,totq=1,totc=1,last[N],tot=1,que;
int cnt[N],to[N][20],depth[N],val[N],st[N],tp=0,v[N],w[N],fi[N],tote=0;
long long now=0,ans[N];
bool vis[N];

struct E{
    int to,next;
}e[N*2];

struct Q{
    int l,r,ti,id;
}q[N];

struct C{
    int u,to,pre;
}c[N];

inline void addedge(int x,int y){
    e[tote].next=fi[x];
    e[tote].to=y;
    fi[x]=tote++;
}
bool cmp(Q a,Q b){
    if(block[a.l]!=block[b.l])
        return block[a.l]<block[b.l];
    else if(block[a.r]!=block[b.r])
        return block[a.r]<block[b.r];
    return a.ti<b.ti;
}

void dfs(int now,int fa,int dep){
    int si=tp;
    to[now][0]=fa;
    depth[now]=dep;
    for(int i=fi[now];i!=-1;i=e[i].next){
        if(e[i].to!=fa){
            dfs(e[i].to,now,dep+1);
            if(tp-1-si>=blocks){
                while(tp-1!=si)
                    block[st[--tp]]=tot;
                tot++;
            }
        }
    }
    st[tp++]=now;
}

inline void getlca(){
    for(int i=1;i<=18;i++)
        for(int j=1;j<=n;j++)
            to[j][i]=to[to[j][i-1]][i-1];
}

inline int lca(int a,int b){
    if(depth[a]>depth[b])
        swap(a,b);
    for(int i=18;i>=0;i--)
        if(depth[to[b][i]]>=depth[a])
            b=to[b][i];
    if(a==b) return a;
    for(int i=18;i>=0;i--){
        if(to[a][i]!=to[b][i]){
            a=to[a][i];
            b=to[b][i];
        }
    }
    return to[a][0];
}

inline void modify(int add,int del,int x){
    int po=c[x].u;
    if(po==0) return ;
    if(vis[po]){
        now-=1ll*v[del]*w[cnt[del]--];
        now+=1ll*v[add]*w[++cnt[add]];
    }
    val[po]=add;
}

inline void change(int u,int k){
    while(u!=k){
        if(vis[u])
            now-=1ll*v[val[u]]*w[cnt[val[u]]--];
        else
            now+=1ll*v[val[u]]*w[++cnt[val[u]]];
        vis[u]=!vis[u];
        u=to[u][0];
    }
}

int main(){
    memset(fi,-1,sizeof(fi));
    scanf("%d%d%d",&n,&m,&que);
    for(int i=1;i<=m;i++)scanf("%d",&v[i]);
    for(int i=1;i<=n;i++)scanf("%d",&w[i]);
    for(int i=0,ta,tb;i<n-1;i++){
        scanf("%d%d",&ta,&tb);
        addedge(ta,tb);
        addedge(tb,ta);
    }
    blocks=pow(n,0.6666);
    for(int i=1;i<=n;i++){
        scanf("%d",&val[i]);
        last[i]=val[i];
    }

    for(int i=1,ta;i<=que;i++){
        scanf("%d",&ta);
        if(ta){
            scanf("%d%d",&q[totq].l,&q[totq].r);
            q[totq].id=totq;
            q[totq++].ti=totc;
        }
        else{
            scanf("%d%d",&c[totc].u,&c[totc].to);
            c[totc].pre=last[c[totc].u];
            last[c[totc].u]=c[totc++].to;
        }
    }

    dfs(1,1,1);
    while(tp) block[st[--tp]]=tot-1;
    getlca();
    sort(q+1,q+totq,cmp);

    for(int i=1,l=1,r=1,t=0;i<totq;i++){
        for(;t<q[i].ti;t++) modify(c[t].to,c[t].pre,t);
        for(;t>q[i].ti;t--) modify(c[t-1].pre,c[t-1].to,t-1);
        if(l!=q[i].l){
            int lc=lca(l,q[i].l);
            change(l,lc);
            change(q[i].l,lc);
        }
        if(r!=q[i].r){
            int lc=lca(r,q[i].r);
            change(r,lc);
            change(q[i].r,lc);
        }

        int zx=lca(q[i].l,q[i].r);
        now+=1ll*v[val[zx]]*w[++cnt[val[zx]]];
        ans[q[i].id]=now;
        now-=1ll*v[val[zx]]*w[cnt[val[zx]]--];

        l=q[i].l; r=q[i].r;
    }
    for(int i=1;i<totq;i++) printf("%lld\n",ans[i]);
}