BZOJ3052:[WC2013]糖果公园(树上莫队)

Description

Input

Output

Sample Input

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

Sample Output

84
131
27
84

HINT

Solution

一个讲解

还是改成括号序的写法吧……感觉好理解还好码……

一开始插入和删除函数是像下面这样分开写的，$vis$函数加加减减不知道哪里错了……如果有大爷看出来哪里错了和我说一声啊QAQ

 void Ins(int x)
 {
     vis[x]++;
     if (vis[x]==) Delicacy-=v[c[x]]*w[Keg[c[x]]], Keg[c[x]]--;
     else Keg[c[x]]++, Delicacy+=v[c[x]]*w[Keg[c[x]]];
 }

 void Del(int x)
 {
     vis[x]--;
     if (vis[x]==) Keg[c[x]]++, Delicacy+=v[c[x]]*w[Keg[c[x]]];
     else Delicacy-=v[c[x]]*w[Keg[c[x]]], Keg[c[x]]--;
 }

 void Recov(int x,int val)
 {
     if (vis[x]==) Del(x), c[x]=val, Ins(x);
     c[x]=val;
 }

后来改成天下第一的写法对$vis$搞异或就过了……至今不知道为什么……

还有我才发现我二轮的树上莫队写的是对的啊……只不过$cmp$函数写错真的太开心了……

虽然多那30分也无济于事就是了……

话说我二轮的树上莫队既然是对的那我2月份写的那一发树分块糖果公园为什么T成狗啊

Code

 #include<iostream>
 #include<cstdio>
 #include<cmath>
 #include<algorithm>
 #define N (200009)
 #define LL long long
 using namespace std;

 struct Que{int l,r,t,num,flag; LL ans;}Q[N];
 struct Mdf{int pre,nxt,pos;}M[N];
 struct Edge{int to,next;}edge[N<<];
 LL Delicacy,v[N],w[N];
 int n,m,q,x,y,opt,l=,r,flag,Q_num,M_num,Time,dfs_num;
 int Keg[N*],vis[N],ID[N],a[N];
 int Fir[N],Sec[N],f[N][],Depth[N],c[N];
 int head[N],num_edge;
 bool cmp1(Que a,Que b)
 {
     if (ID[a.l]==ID[b.l])
         return ID[a.r]==ID[b.r]?a.t<b.t:ID[a.r]<ID[b.r];
     else return ID[a.l]<ID[b.l];
 }
 bool cmp2(Que a,Que b) {return a.num<b.num;}

 void add(int u,int v)
 {
     edge[++num_edge].to=v;
     edge[num_edge].next=head[u];
     head[u]=num_edge;
 }

 void Ins(int x)
 {
     if (vis[x]) Delicacy-=v[c[x]]*w[Keg[c[x]]], Keg[c[x]]--;
     else Keg[c[x]]++, Delicacy+=v[c[x]]*w[Keg[c[x]]];
     vis[x]^=;
 }

 void Recov(int x,int val)
 {
     if (vis[x]) Ins(x),c[x]=val,Ins(x);
     else c[x]=val;
 }

 void MoQueue(int num)
 {
     while (Time<Q[num].t) Recov(M[Time+].pos,M[Time+].nxt), Time++;
     while (Time>Q[num].t) Recov(M[Time].pos,M[Time].pre), Time--;
     while (l<Q[num].l) Ins(a[l++]);
     while (l>Q[num].l) Ins(a[--l]);
     while (r<Q[num].r) Ins(a[++r]);
     while (r>Q[num].r) Ins(a[r--]);
     Q[num].ans=Delicacy;
     if (Q[num].flag)
     {
         Ins(Q[num].flag);
         Q[num].ans=Delicacy;
         Ins(Q[num].flag);
     }
 }

 void DFS(int x,int fa)
 {
     f[x][]=fa;
     for (int i=; i<=; ++i) f[x][i]=f[f[x][i-]][i-];
     Depth[x]=Depth[fa]+; Fir[x]=++dfs_num;
     a[dfs_num]=x;
     for (int i=head[x]; i; i=edge[i].next)
         if (edge[i].to!=fa) DFS(edge[i].to,x);
     Sec[x]=++dfs_num; a[dfs_num]=x;
 }

 int LCA(int x,int y)
 {
     if (Depth[x]<Depth[y]) swap(x,y);
     for (int i=; i>=; --i)
         if (Depth[f[x][i]]>=Depth[y]) x=f[x][i];
     if (x==y) return x;
     for (int i=; i>=; --i)
         if (f[x][i]!=f[y][i]) x=f[x][i], y=f[y][i];
     return f[x][];
 }
 int main()
 {
     scanf("%d%d%d",&n,&m,&q);
     int unit=pow(n,2.0/3.0);
     for (int i=; i<=(n*); ++i) ID[i]=(i-)/unit+;
     for (int i=; i<=m; ++i) scanf("%lld",&v[i]);
     for (int i=; i<=n; ++i) scanf("%lld",&w[i]);
     for (int i=; i<=n-; ++i)
         scanf("%d%d",&x,&y), add(x,y), add(y,x);
     for (int i=; i<=n; ++i) scanf("%d",&c[i]);
     DFS(,);
     for (int i=; i<=q; ++i)
     {
         scanf("%d%d%d",&opt,&x,&y);
         if (opt==)
         {
             if (Fir[x]>Fir[y]) swap(x,y);
             int lca=LCA(x,y);
             if (lca==x) x=Fir[x], y=Fir[y], flag=;
             else x=Sec[x], y=Fir[y], flag=lca;
             Q[++Q_num]=(Que){x,y,M_num,i,flag,};
         }
         else M[++M_num]=(Mdf){c[x],y,x}, c[x]=y;
     }
     for (int i=M_num; i>=; --i)
         c[M[i].pos]=M[i].pre;
     sort(Q+,Q+Q_num+,cmp1);
     for (int i=; i<=Q_num; ++i)
         MoQueue(i);
     sort(Q+,Q+Q_num+,cmp2);
     for (int i=; i<=Q_num; ++i)
         printf("%lld\n",Q[i].ans);
 }