糖果公园

糖果公园

题目描述
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 100010using namespace std;int n,m,Q,v[N],w[N],blo[N],col[N],dep[N];int k,la[N],ff[N*2],fa[N],cnt,num,q[N];int tong[N],flag[N],tmp,tot,B,pre[N],last[N];int size[N],next[N],top[N];ll res[N],ans;struct node{int a,b;}map[N*2],s[N];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[N];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;}