BZOJ.3052.[WC2013]糖果公园(树上莫队 带修改莫队)

题目链接 BZOJ

当然哪都能交(都比在BZOJ交好)，比如UOJ #58

//67376kb	27280ms
//树上莫队+带修改莫队 模板题
#include <cmath>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <algorithm>
//#define gc() getchar()
#define MAXIN 400000
#define gc() (SS==TT&&(TT=(SS=IN)+fread(IN,1,MAXIN,stdin),SS==TT)?EOF:*SS++)
typedef long long LL;
const int N=1e5+5,M=1e6+3;

int n,m,Enum,H[N],nxt[N<<1],to[N<<1],val[M],W[N],col[N],las[N],tm[M];
int fa[N],dep[N],sz[N],son[N],top[N],Index,in[N],out[N],seq[N<<1];
LL Now,Ans[M];
bool vis[N];
char IN[MAXIN],*SS=IN,*TT=IN;
struct Queries
{
	int l,r,lca,bl,br,tm,id;
	bool operator <(const Queries &x)const
	{
		if(bl!=x.bl) return bl<x.bl;//能优化15%？
		if(br!=x.br) return bl&1?br<x.br:br>x.br;
		return (bl^br)&1?tm<x.tm:tm>x.tm;
//		if(bl==x.bl) return br==x.br?tm<x.tm:br<x.br;
//		return bl<x.bl;
	}
}q[M];
struct Modify
{
	int pos,val,bef;
}qm[M];

inline int read()
{
	int now=0;register char c=gc();
	for(;!isdigit(c);c=gc());
	for(;isdigit(c);now=now*10+c-'0',c=gc());
	return now;
}
inline void AddEdge(int u,int v)
{
	to[++Enum]=v, nxt[Enum]=H[u], H[u]=Enum;
	to[++Enum]=u, nxt[Enum]=H[v], H[v]=Enum;
}
inline int LCA(int u,int v)
{
	while(top[u]!=top[v]) dep[top[u]]>dep[top[v]]?u=fa[top[u]]:v=fa[top[v]];
	return dep[u]>dep[v]?v:u;
}
void DFS1(int x)
{
	int mx=0; sz[x]=1;
	for(int v,i=H[x]; i; i=nxt[i])
		if((v=to[i])!=fa[x])
		{
			fa[v]=x, dep[v]=dep[x]+1, DFS1(v), sz[x]+=sz[v];
			if(mx<sz[v]) mx=sz[v], son[x]=v;
		}
}
void DFS2(int x,int tp)
{
	seq[in[x]=++Index]=x, top[x]=tp;
	if(son[x])
	{
		DFS2(son[x],tp);
		for(int i=H[x]; i; i=nxt[i])
			if(to[i]!=fa[x]&&to[i]!=son[x]) DFS2(to[i],to[i]);
	}
	seq[out[x]=++Index]=x;
}
void Calc(int p)
{
	vis[p] ? Now-=1ll*W[tm[col[p]]--]*val[col[p]] : Now+=1ll*W[++tm[col[p]]]*val[col[p]];
	vis[p]^=1;
}
void Change(int p,int v)
{
	if(vis[p]) Calc(p), col[p]=v, Calc(p);
	else col[p]=v;
}

int main()
{
	n=read(), m=read(); int Q=read();
	for(int i=1; i<=m; ++i) val[i]=read();
	for(int i=1; i<=n; ++i) W[i]=read();
	for(int i=1; i<n; ++i) AddEdge(read(),read());
	for(int i=1; i<=n; ++i) col[i]=las[i]=read();
	DFS1(1), DFS2(1,1);
	int qcnt=0, mcnt=0, size=pow(n,2.0/3.0);
	for(int i=1,x,y,w; i<=Q; ++i)
		if(!read()) qm[++mcnt]=(Modify){x=read(),y=read(),las[x]},las[x]=y;
		else
		{
			x=read(), y=read();
//			if(x==y) {q[++qcnt].lca=-1, Ans[qcnt]=...; continue;}
			w=LCA(x,y);
			if(in[x]>in[y]) std::swap(x,y);
			if(x==w) q[++qcnt]=(Queries){in[x],in[y],0,in[x]/size,in[y]/size,mcnt,qcnt};
			else q[++qcnt]=(Queries){out[x],in[y],w/*in[w]*/,out[x]/size,in[y]/size,mcnt,qcnt};
		}
	std::sort(q+1,q+1+qcnt);
	for(int i=1,t=0,l=1,r=0,ql,qr,qt; i<=qcnt; ++i)
	{
		ql=q[i].l, qr=q[i].r, qt=q[i].tm;
		while(t<qt) ++t,Change(qm[t].pos,qm[t].val);
		while(t>qt) Change(qm[t].pos,qm[t].bef),--t;
		if(ql==qr) {Ans[q[i].id]=1ll*val[col[seq[q[i].l]]]*W[1]; continue;}
		while(l<ql) Calc(seq[l++]);
		while(l>ql) Calc(seq[--l]);
		while(r<qr) Calc(seq[++r]);
		while(r>qr) Calc(seq[r--]);
		if(q[i].lca) Calc(q[i].lca);//Calc(seq[in[lca]]) = Calc(lca)
		Ans[q[i].id]=Now;
		if(q[i].lca) Calc(q[i].lca);
	}
	for(int i=1; i<=qcnt; ++i) printf("%lld\n",Ans[i]);

	return 0;
}