P4074 [WC2013]糖果公园 (树上带修莫队)
题目链接:P4074 [WC2013]糖果公园
代码
//#include<bits/stdc++.h>
#include<iostream>
#include<algorithm>
#include<set>
#include<vector>
#include<list>
#include<queue>
#include<cmath>
#include<cstring>
#include<sstream>
#include<cstdio>
#include<ctime>
#include<map>
#include<bitset>
#include<stack>
#include<string>
using namespace std;
//#define getchar() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++)
// char buf[1<<21],*p1=buf,*p2=buf;
#define pii pair<int,int>
#define pll pair<ll,ll>
#define ll long long
#define pb push_back
#define ull unsigned ll
#define rg register
#define bug puts("bug");
const long long INF=0x3f3f3f3f3f3f3f3fLL;
const int inf = 0x3f3f3f3f;
const int maxn = 1e5+10;
const int mod = 1e9+7;
const double eps = 1e-6;
int di[8][2] = {{-1,0},{1,0},{0,-1},{0,1},{-1,-1},{-1,1},{1,-1},{1,1}};
inline ll rd()
{
ll x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=(x<<3)+(x<<1)+(ch^48);ch=getchar();}
return x*f;
}
int n,m,k;
int V[maxn],W[maxn],C[maxn];
vector<int>ve[maxn];
int siz[maxn],top[maxn],son[maxn],f[maxn];
int deep[maxn];
vector<int>oula(1,0);//欧拉序
int st[maxn],ed[maxn];
ll ans[maxn];
void dfs1(int x,int fa)
{
st[x]=oula.size();
oula.pb(x);
deep[x]=deep[fa]+1;
siz[x]=1;
f[x]=fa;
for(int& y:ve[x])
{
if(y!=fa)
{
dfs1(y,x);
siz[x]+=siz[y];
if(siz[y]>siz[son[x]])son[x]=y;
}
}
ed[x]=oula.size();
oula.pb(x);
}
void dfs2(int x,int t)
{
top[x]=t;
if(son[x])dfs2(son[x],t);
for(int &y:ve[x])
{
if(y!=son[x]&&y!=f[x])dfs2(y,y);
}
}
int lca(int x,int y)
{
while(top[x]!=top[y])
{
if(deep[top[x]]<deep[top[y]])swap(x,y);
x=f[top[x]];
}
return deep[x]>deep[y]?y:x;
}
int LCA[maxn];
int T;
int cntq,cntc;
struct node{
int id,l,r,t;
void read()
{
id=cntq;
t=cntc;
l=rd();
r=rd();
if(st[l]>st[r])swap(l,r);
int k=lca(l,r);
if(k==l||l==r)
{
l=st[l];
r=st[r];
}
else
{
l=ed[l];
r=st[r];
LCA[id]=k;
}
}
bool friend operator<(const node& n1,const node& n2)
{
if(n1.l/T!=n2.l/T)return n1.l<n2.l;
if(n1.r/T!=n2.r/T)return n1.r<n2.r;
return n1.t<n2.t;
}
}q[maxn];
struct node1{
int id,p,k;
void read()
{
id=cntc;
p=rd();
k=rd();
}
}c[maxn];
ll now;
int v[maxn];
int cnt[maxn];
void Add(int x,int flag=1)
{
if(flag)x=C[x];
cnt[x]++;
now+=1ll*V[x]*W[cnt[x]];
}
void Sub(int x,int flag=1)
{
if(flag)x=C[x];
now-=1ll*V[x]*W[cnt[x]];
cnt[x]--;
}
void add(int x)
{
x=oula[x];
if(v[x])Sub(x);
else Add(x);
v[x]^=1;
}
bool in(int x,int i)
{
return x>=q[i].l&&x<=q[i].r;
}
void move(int t,int i)
{
int x=c[t].p;
if(in(st[x],i)^in(ed[x],i))
{
Sub(C[c[t].p],0);
Add(c[t].k,0);
}
swap(c[t].k,C[c[t].p]);
}
int main()
{
n=rd(),m=rd(),k=rd();
for(int i=1;i<=m;i++)V[i]=rd();
for(int i=1;i<=n;i++)W[i]=rd();
for(int i=1;i<n;i++)
{
int u=rd(),v=rd();
ve[u].pb(v);
ve[v].pb(u);
}
dfs1(1,0);
dfs2(1,1);
for(int i=1;i<=n;i++)C[i]=rd();
for(int i=1;i<=k;i++)
{
int op=rd();
if(op==1)
{
q[++cntq].read();
}
else
{
c[++cntc].read();
}
}
T=pow(n,2.0/3);
sort(q+1,q+1+cntq);
int l=1,r=0,t=0;
for(int i=1;i<=cntq;i++)
{
while(l<q[i].l)add(l++);
while(l>q[i].l)add(--l);
while(r<q[i].r)add(++r);
while(r>q[i].r)add(r--);
while(t>q[i].t)move(t--,i);
while(t<q[i].t)move(++t,i);
if(LCA[q[i].id])Add(LCA[q[i].id]);
ans[q[i].id]=now;
if(LCA[q[i].id])Sub(LCA[q[i].id]);
}
for(int i=1;i<=cntq;i++)printf("%lld\n",ans[i]);
}