糖果公园

糖果公园

题目描述
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[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;
}