【BZOJ 3052】 [wc2013]糖果公园

3052: [wc2013]糖果公园

Time Limit: 250 Sec Memory Limit: 512 MB

Submit: 481 Solved: 176

[Submit][Status]

Description

Input

Output

Sample Input

Sample Input

Sample Output

84

131

27

84

HINT

带修改的树上莫队。

完成
【BZOJ 3757】（不带修改的树上莫队）之后，这道题就好理解了。

与bzoj3757一样，先对树分块，把询问以块和查询时间为关键字排序，关键就是转移了。

(curV,curU)到(targetV,targetU)的转移与bzoj3757一致，那么修改怎么处理呢？

预处理出每一个询问操作的前一个修改操作是第几个。

分两种情况：

1.输入中cur比target靠前，那么我们只要把从cur后面到target之间的所有修改处理一次就可以。

2.输入中cur比target靠后，那么我们需要预处理出每一个修改操作中修改之前的颜色是pre，把cur到target之间的修改处

理一次，与上面不同的是现在要把当前颜色修改成pre。

综上，只要在(curV,curU)到(targetV,targetU)转移之前对时间进行上述转移即可。

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#define M 100005
#define LL long long
using namespace std;
LL v[M],w[M],Ans[M],ans=0LL;
int k,QQ1,dfs_clock=0,dep[M],dfn[M],be[M],b[25],f[M][22],h[M],tot=0,cnt=0,Q,QQ,n,m,vi[M],s[M],top=0,c[M],num[M],pre[M];
struct edge
{
int y,ne;
}e[M*2];
struct Q1
{
int x,y,k,pre;
}q1[M],qq[M];
struct Query
{
int x,y,id,t;
}q[M];
void r(int &tmp)
{
tmp=0;
char ch=getchar();
for (;ch<'0'||ch>'9';ch=getchar());
for (;ch>='0'&&ch<='9';ch=getchar())
tmp=(tmp*10)+ch-'0';
}
void R(LL &tmp)
{
tmp=0LL;
char ch=getchar();
for (;ch<'0'||ch>'9';ch=getchar());
for (;ch>='0'&&ch<='9';ch=getchar())
tmp=(tmp*10)+ch-'0';
}
void Addedge(int x,int y)
{
e[++tot].y=y;
e[tot].ne=h[x];
h[x]=tot;
}
void dfs(int x)
{
dfn[x]=++dfs_clock;
int bottom=top;
for (int i=h[x];i;i=e[i].ne)
{
int y=e[i].y;
if (f[x][0]==y) continue;
f[y][0]=x;
dep[y]=dep[x]+1;
dfs(y);
if (top-bottom>=k)
{
cnt++;
while (top!=bottom)
be[s[top--]]=cnt;
}
}
s[++top]=x;
}
void prepare()
{
b[0]=1;
for (int i=1;i<=16;i++)
b[i]=b[i-1]<<1;
for (int j=1;j<=16;j++)
for (int i=1;i<=n;i++)
f[i][j]=f[f[i][j-1]][j-1];
}
bool cmp(Query a,Query b)
{
if (be[a.x]==be[b.x])
{
if (be[a.y]==be[b.y])
return a.t<b.t;
return be[a.y]<be[b.y];
}
return be[a.x]<be[b.x];
}
void Calc(int x)
{
if (vi[x])
ans=ans-(v[c[x]]*(w[num[c[x]]--]));
else
ans=ans+(v[c[x]]*(w[++num[c[x]]]));
vi[x]^=1;
}
void Modifytime(int x,int y)
{
if (q[x].id<q[y].id)
{
for (int i=q[x].id+1;i<=q[y].id;i++)
{
int now=c[qq[i].x],mo=qq[i].y;
if (vi[qq[i].x])
{
Calc(qq[i].x);
c[qq[i].x]=mo;
Calc(qq[i].x);
}
else c[qq[i].x]=mo;
}
return;
}
for (int i=q[x].id;i>q[y].id;i--)
{
int now=c[qq[i].x],mo=qq[i].pre;
if (vi[qq[i].x])
{
Calc(qq[i].x);
c[qq[i].x]=mo;
Calc(qq[i].x);
}
else c[qq[i].x]=mo;
}
}
void Move(int &x,int deep)
{
for (int i=16;i>=0;i--)
if (dep[f[x][i]]>=deep)
x=f[x][i];
}
int Getlca(int x,int y)
{
if (dep[x]>dep[y]) swap(x,y);
Move(y,dep[x]);
if (x==y) return x;
for (int i=16;i>=0;i--)
if (f[x][i]!=f[y][i])
x=f[x][i],y=f[y][i];
return f[x][0];
}
void Solve(int x,int y)
{
while (x!=y)
{
if (dep[x]<dep[y])  swap(x,y);
Calc(x);
x=f[x][0];
}
}
int main()
{
r(n),r(m),r(QQ);
k=sqrt(n);
for (int i=1;i<=m;i++)
R(v[i]);
for (int i=1;i<=n;i++)
R(w[i]);
for (int i=1;i<n;i++)
{
int x,y;
r(x),r(y);
Addedge(x,y),Addedge(y,x);
}
dep[1]=1;
dfs(1);
while (top)
be[s[top--]]=cnt;
prepare();
for (int i=1;i<=n;i++)
r(c[i]),pre[i]=c[i];
for (int i=1;i<=QQ;i++)
{
int t,x,y;
r(t),r(x),r(y);
if (!t)
{
QQ1++;
qq[QQ1].x=x,qq[QQ1].y=y;
qq[QQ1].pre=pre[x];
pre[x]=y;
}
else
{
Q++;
if (dfn[x]>dfn[y])
swap(x,y);
q[Q].x=x,q[Q].y=y;
q[Q].id=QQ1;
q[Q].t=Q;
}
}
sort(q+1,q+1+Q,cmp);
ans=0LL;
q[0].id=0,q[0].x=q[0].y=1;
for (int i=1;i<=Q;i++)
{
Modifytime(i-1,i);
Solve(q[i-1].x,q[i].x);
Solve(q[i-1].y,q[i].y);
int a=Getlca(q[i].x,q[i].y);
Calc(a);
Ans[q[i].t]=ans;
Calc(a);
}
for (int i=1;i<=Q;i++)
printf("%lld\n",Ans[i]);
return 0;
}