参考:http://blog.csdn.net/lych_cys/article/details/50845832
把树变成dfs括号序的形式,注意这个是不包含lca的(除非lca是两点中的一个)
然后把询问按照所属块一序,r二序,t三序排序(注意a和b数组的同名变量意思不一样),对于每个询问处理修改,时间正流或者逆流,修改答案时用vis数组记录这个位置是否在答案中被统计过来决定加减。对于修改序列的操作,如果他没有被统计在答案里就直接修改,否则修改答案。对lca特殊处理
#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
const int N=200005;
int n,m,q,u[N],v[N],w[N],h[N],cnt,c[N],la[N],kuai,f[N],g[N],dfn,bl[N],cnta,cntb;
int de[N],fa[N],fr[N],si[N],hs[N],rl[N];
long long ans[N],sum;
bool vis[N];
struct qw
{
int ne,to;
}e[N<<1];
struct qwe
{
int l,r,t,rl;
}a[N],b[N];//对于b,l是位置,r是修改为的颜色,t是上一个颜色
bool cmp(qwe x,qwe y)
{
return bl[x.l]<bl[y.l]||bl[x.l]==bl[y.l]&&bl[x.r]<bl[y.r]||bl[x.l]==bl[y.l]&&bl[x.r]==bl[y.r]&&x.t<y.t;
}
int read()
{
int r=0;
char p=getchar();
while(p>‘9‘||p<‘0‘)
p=getchar();
while(p>=‘0‘&&p<=‘9‘)
{
r=r*10+p-48;
p=getchar();
}
return r;
}
void add(int u,int v)
{
cnt++;
e[cnt].ne=h[u];
e[cnt].to=v;
h[u]=cnt;
}
void dfs1(int u,int fat)
{
fa[u]=fat;
de[u]=de[fat]+1;
si[u]=1;
f[u]=++dfn;
rl[dfn]=u;
for(int i=h[u];i;i=e[i].ne)
if(e[i].to!=fat)
{
dfs1(e[i].to,u);
si[u]+=si[e[i].to];
if(si[e[i].to]>si[hs[u]])
hs[u]=e[i].to;
}
g[u]=++dfn;
rl[dfn]=u;
}
void dfs2(int u,int top)
{
fr[u]=top;
if(!hs[u])
return;
dfs2(hs[u],top);
for(int i=h[u];i;i=e[i].ne)
if(e[i].to!=fa[u]&&e[i].to!=hs[u])
dfs2(e[i].to,e[i].to);
}
int lca(int u,int v)
{
for(;fr[u]!=fr[v];de[fr[u]]>de[fr[v]]?u=fa[fr[u]]:v=fa[fr[v]]);
return de[u]>de[v]?v:u;
}
void add(int x)
{
if(vis[x])
sum-=(long long)v[c[x]]*w[u[c[x]]--];
else
sum+=(long long)v[c[x]]*w[++u[c[x]]];
vis[x]^=1;
}
void change(int x,int y)
{
if(vis[x])
{
add(x);
c[x]=y;
add(x);
}
else
c[x]=y;
}
int main()
{
n=read(),m=read(),q=read();
for(int i=1;i<=m;i++)
v[i]=read();
for(int i=1;i<=n;i++)
w[i]=read();
for(int i=1;i<n;i++)
{
int x=read(),y=read();
add(x,y);add(y,x);
}
for(int i=1;i<=n;i++)
c[i]=la[i]=read();
kuai=(int)pow(n,2.0/3);
dfs1(1,0);
dfs2(1,1);
for(int i=1;i<=dfn;i++)
bl[i]=(i-1)/kuai;
while(q--)
{
int t=read(),l=read(),r=read();
if(t==0)
{
b[++cntb].l=l;
b[cntb].t=la[l];
la[l]=b[cntb].r=r;
}
else
{
if(f[l]>f[r])
swap(l,r);
a[++cnta].r=f[r];
a[cnta].rl=cnta;
a[cnta].t=cntb;//cout<<lca(l,r)<<endl;
a[cnta].l=(lca(l,r)==l)?f[l]:g[l];
}
}
sort(a+1,a+cnta,cmp);
int l=1,r=0,t=1;
for(int i=1;i<=cnta;i++)
{
for(;t<=a[i].t;t++)
change(b[t].l,b[t].r);
for(;t>a[i].t;t--)
change(b[t].l,b[t].t);
while(l>a[i].l)
add(rl[--l]);
while(l<a[i].l)
add(rl[l++]);
while(r>a[i].r)
add(rl[r--]);
while(r<a[i].r)
add(rl[++r]);
int x=rl[l],y=rl[r],tmp=lca(x,y);//cout<<x<<" "<<y<<" "<<tmp<<endl;
if(x!=tmp&&y!=tmp)
{
add(tmp);
ans[a[i].rl]=sum;
add(tmp);
}
else
ans[a[i].rl]=sum;
}
for(int i=1;i<=cnta;i++)
printf("%lld\n",ans[i]);
return 0;
}
/*
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
*/