bzoj3052 糖果公园 树上莫队+修改

http://www.elijahqi.win/archives/378
Description

Input

Sample Input
4 3 5
1 9 2
7 6 5 1
2 3
3 1
3 4
1 2 3 2
1 1 2
1 4 2
0 2 1
1 1 2
1 4 2
Sample Output
84
131
27
84
HINT

另外我把数据重敲了，放在sample上，原题是图片。。

昨天刚写完一个树上莫队，今天记忆尤新，这个主要是加了修改操作，具体可以参考我blog bzoj2120

注意last 得赋予初值，否则肯定有问题，初值就是c[]一开始的；

在昨天基础上稍微改良了下dfs直接返回top的值即可

今天也再次明确了一下树上莫队的排序条件，左端点在同一块时，按照右端点的dfs序，从小到大排序，为了满足均摊o(1)的转移时间

左端点块小，大概的深度就深（不一定深） 右端从小到大，保证转移复杂度低。

bzoj200s时限，建议可以去uoj 58题交一交，再去bzoj，怕引起民愤hh。

#include<cstdio>
#include<cmath>
#include<algorithm>
#define LL long long
#define N 110000
int n,m,q1,c[N],Log[N],f[N],dfn[N],num,low[N],h[N],fa[N][20],n1,block_num,bl[N],last[N];
bool visit[N];
LL ans[N],ans1,v[N],w[N];
inline int read(){
    int x=0;char ch=getchar();
    while (ch<'0'||ch>'9') ch=getchar();
    while (ch<='9'&&ch>='0'){x=x*10+ch-'0';ch=getchar();}
    return x;
}
struct node{
    int y,next;
}data[N<<1];
struct node1{
    int l,r,id,time;
}q[N];
struct node2{
    int pos,pre,color;
}d[N];
inline void swap(int &x,int &y){
    x^=y;y^=x;x^=y;
}
int stack[N],top;
int dfs(int x){
    int size=0;dfn[x]=++num;
    for (int i=1;i<=Log[low[x]];++i) fa[x][i]=fa[fa[x][i-1]][i-1];
    for (int i=h[x];i;i=data[i].next){
        int y=data[i].y;
        if (y==fa[x][0]) continue;
        fa[y][0]=x;
        low[y]=low[x]+1;
        size+=dfs(y);
        if (size>=n1){
            block_num++;
            while (top) bl[stack[top--]]=block_num;
        }
    }
    stack[++top]=x;
    return top;
}
inline int lca(int x,int y){
    if (low[x]<low[y]) swap(x,y);
    int dis=low[x]-low[y];
    for (int i=0;i<=Log[dis];++i) if ((1<<i)&dis) x=fa[x][i];
    if (x==y) return x;
    for (int i=Log[n];i>=0;--i) if (fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
inline void change(int x,int y){
    if (visit[x]){
        ans1-=w[f[c[x]]]*v[c[x]];f[c[x]]--;
        c[x]=y;
        f[c[x]]++;ans1+=w[f[c[x]]]*v[c[x]];
    }else c[x]=y;
}
inline bool cmp(node1 a,node1 b){
    if(bl[a.l]==bl[b.l]){
        return dfn[a.r]==dfn[b.r]?a.time<b.time:dfn[a.r]<dfn[b.r];
    }else return bl[a.l]<bl[b.l];
}
inline void reserve(int x){
    if (visit[x]) ans1-=w[f[c[x]]]*v[c[x]],f[c[x]]--;else f[c[x]]++,ans1+=w[f[c[x]]]*v[c[x]];
    visit[x]^=1;
}
inline void solve(int x,int y){
    while (x!=y) if (low[y]>low[x]) reserve(y),y=fa[y][0];else reserve(x),x=fa[x][0];
}
int main(){
    //freopen("bzoj3052.in","r",stdin);
//  freopen("bzoj3052.out","w",stdout);
    n=read();m=read();q1=read();n1=sqrt(n);
    for (int i=1;i<=m;++i) v[i]=read();
    for (int i=1;i<=n;++i) w[i]=read();
    int x,y;
    for (int i=1;i<n;++i){
        x=read();y=read();
        data[++num].y=y;data[num].next=h[x];h[x]=num;
        data[++num].y=x;data[num].next=h[y];h[y]=num;
    }
    Log[0]=-1;
    for (int i=1;i<=n;++i) Log[i]=Log[i>>1]+1;
    for (int i=1;i<=n;++i) c[i]=read(),last[i]=c[i];
    num=0;dfs(1);++block_num;while (top) bl[stack[top--]]=block_num;
    int type,cnt=0;num=0;
    for (int i=1;i<=q1;++i){
        type=read();x=read();y=read();
        if (type){
            q[++num].l=x;q[num].r=y;q[num].id=num;q[num].time=cnt;
            if (bl[x]>bl[y]) swap(q[num].l,q[num].r);
        }else{
            d[++cnt].pre=last[x];last[x]=y;d[cnt].color=y;d[cnt].pos=x;

        }
    }
    std::sort(q+1,q+num+1,cmp);
    //for (int i=1;i<=num;++i) printf("%d %d\n",q[i].l,q[i].r);
//  printf("%d",q[1].time);
//  for (int i=1;i<=n;++i) printf("%d ",c[i]);
/*  for (int i=1;i<=n;++i){
        for (int j=0;j<=3;++j) printf("%d ",fa[i][j]);
        printf("\n");
    }*/
//  for (int i=1;i<=n;++i) printf("%d ",low[i]);
//  printf("%d %d\n",d[1].color,d[1].pre);
    for (int i=1;i<=q[1].time;++i) c[d[i].pos]=d[i].color;
    solve(q[1].l,q[1].r);
    type=lca(q[1].l,q[1].r);
    ans[q[1].id]=ans1+v[c[type]]*w[f[c[type]]+1];
    for (int i=2;i<=num;++i){
        for (int j=q[i-1].time+1;j<=q[i].time;++j) change(d[j].pos,d[j].color);
        for (int j=q[i-1].time;j>q[i].time;--j) change(d[j].pos,d[j].pre);
        solve(q[i-1].l,q[i].l);solve(q[i-1].r,q[i].r);
        type=lca(q[i].l,q[i].r);
        ans[q[i].id]=ans1+w[f[c[type]]+1]*v[c[type]];
        //printf("%d %d\n",ans1+w[f[c[type]]+1]*v[c[type]],q[i].id);
    }
    for (int i=1;i<=num;++i) printf("%lld\n",ans[i]); 
    return 0;
}