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]==2) 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]==1) 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]==1) 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<<1];
LL Delicacy,v[N],w[N];
int n,m,q,x,y,opt,l=1,r,flag,Q_num,M_num,Time,dfs_num;
int Keg[N*5],vis[N],ID[N],a[N];
int Fir[N],Sec[N],f[N][18],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]^=1;
}

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