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