糖果公园

糖果公园

题目描述

http://www.lydsy.com/JudgeOnline/problem.php?id=3052

题解

树上莫队。把树分块，按照blo[x]，blo[y]，time三个关键字依次排序。对于每次询问操作，先把时间倒流，如果这个点被跑过就作出修改。再在树上跳一跳，把每次操作的起点和上次操作的起点的lca的路径上的点的存在性取反，并修改答案即可。

块的大小为n^(1/3)（大概是这样跑得快把。。）。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<cmath>
#include<algorithm>
#define ll long long
#define N 100010
using namespace std;
int n,m,Q,v
,w
,blo
,col
,dep
;
int k,la
,ff[N*2],fa
,cnt,num,q
;
int tong
,flag
,tmp,tot,B,pre
,last
;
int size
,next
,top
;
ll res
,ans;
struct node{int a,b;}map[N*2],s
;
struct info{
int x,y,t,id;
bool operator<(const info &p)const{
if(blo[x]!=blo[p.x])return blo[x]<blo[p.x];
if(blo[y]!=blo[p.y])return blo[y]<blo[p.y];
return t<p.t;
}
}t
;

inline int get()
{
char ch;
while(!isdigit(ch=getchar()));
int v=ch-48;
while(isdigit(ch=getchar()))v=v*10+ch-48;
return v;
}

inline void add(int a,int b)
{
map[++k]=(node){a,b};ff[k]=la[a];la[a]=k;
map[++k]=(node){b,a};ff[k]=la[b];la[b]=k;
}

inline void dfs(int x)
{
for(int a=la[x];a;a=ff[a])
if(fa[x]!=map[a].b)
{
dep[map[a].b]=dep[x]+1;
fa[map[a].b]=x;dfs(map[a].b);
if(num>=B){cnt++;while(num)blo[q[num--]]=cnt;}
}
q[++num]=x;
}

inline void bfs()
{
int l=1,r=2;q[1]=1;
while(l<r)
{
int x=q[l];l++;size[x]=1;
for(int a=la[x];a;a=ff[a])
if(fa[x]!=map[a].b)
q[r]=map[a].b,dep[q[r]]=dep[x]+1,r++;
}
for(int i=r-1;i;i--)
{
int x=q[i];size[fa[x]]+=size[x];
if(size[x]>size[next[fa[x]]])next[fa[x]]=x;
}
for(int i=1;i<r;i++)
{
int x=q[i];
if(!flag[x])
for(int j=x;j;j=next[j])top[j]=x,flag[j]=1;
}
memset(flag,0,sizeof(flag));
}

inline int lca(int a,int b)
{
while(top[a]!=top[b])
{
if(dep[top[a]]<dep[top[b]])swap(a,b);
a=fa[top[a]];
}
if(dep[a]<dep[b])return a;
return b;
}

inline void change(int a,int b)
{
if(flag[a])
{
ans-=(ll)w[tong[col[a]]]*v[col[a]];
tong[col[a]]--;tong[b]++;
ans+=(ll)w[tong[b]]*v[b];
}
col[a]=b;
}

inline void reverse(int x)
{
if(flag[x])ans-=(ll)w[tong[col[x]]]*v[col[x]],tong[col[x]]--;
else tong[col[x]]++,ans+=(ll)w[tong[col[x]]]*v[col[x]];
flag[x]^=1;
}

inline void modify_time(int x,int y)
{
if(x==y)return;
if(x<y)for(x++;x<=y;x++)change(s[x].a,s[x].b);
else for(;x>y;x--)change(s[x].a,pre[x]);
}

inline void solve(int x,int y)
{
while(x!=y)
{
if(dep[x]<dep[y])swap(x,y);
reverse(x);x=fa[x];
}
}

inline void prepare()
{
modify_time(0,t[1].t);
solve(t[1].x,t[1].y);
int x=lca(t[1].x,t[1].y);
reverse(x);res[t[1].id]=ans;reverse(x);
}

int main()
{
int tp,c,x,y;
n=get();m=get();Q=get();
B=int(pow(n,2.0/3.0)*0.5);
for(int i=1;i<=m;i++)v[i]=get();
for(int i=1;i<=n;i++)w[i]=get();
for(int i=1;i<n;i++)add(get(),get());
dep[1]=1;dfs(1);bfs();if(!cnt)cnt++;
while(num)blo[q[num--]]=cnt;
for(int i=1;i<=n;i++)
col[i]=get(),last[i]=col[i];
for(int i=1;i<=Q;i++)
{
tp=get();x=get();y=get();
if(!tp)
{
s[++tmp]=(node){x,y};
pre[tmp]=last[x];last[x]=y;
}
else t[++tot]=(info){x,y,tmp,tot};
}
sort(t+1,t+tot+1);if(tot)prepare();
for(int i=2;i<=tot;i++)
{
modify_time(t[i-1].t,t[i].t);
solve(t[i-1].x,t[i].x);
solve(t[i-1].y,t[i].y);
int x=lca(t[i].x,t[i].y);
reverse(x);res[t[i].id]=ans;reverse(x);
}
for(int i=1;i<=tot;i++)printf("%lld\n",res[i]);
return 0;
}