题目大意:给定一棵n个点的树,每个点上有一个汉子或妹子,每人有一个权值,每次询问一条链上选出一对权值相等的男女有多少种选法。(n,q<=10^5)
做法:比较显然的树上莫队,熟悉序列莫队那套理论再推广到树上即可,我的树上莫队好像有点假,我是先树分块,对每块的根跑一遍dfs处理到各个点的链的信息并顺便处理一个端点在这个块内的询问,常数好像有点大,卡了半天常才过……学习了一下dalao的树上莫队,求一个进和出都算一遍的dfs序,链询问转化为区间询问,在区间内但不在链上的会被计算到两次所以被抵消了,感觉很奇妙。
丑陋代码:
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<map>
using namespace std;
char BB[<<],*S=BB;
inline int read()
{
int x;char c;
while((c=*S++)<''||c>'');
for(x=c-'';(c=*S++)>=''&&c<='';)x=x*+c-'';
return x;
}
#define MN 100000
#define K 700
struct edge{int nx,t;}e[MN*+];
struct query{int x,y,id;}Q[MN+];
int a[MN+],b[MN+],cnt,h[MN+],en,fa[MN+],d[MN+],f[MN+],dep[MN+];
int rt,nw,u[MN+],A[MN+],B[MN+];
long long ans,C[MN+];
bool cmp(const query&a,const query&b){return f[a.x]<f[b.x];}
map<int,int> mp;
inline void ins(int x,int y)
{
e[++en]=(edge){h[x],y};h[x]=en;
e[++en]=(edge){h[y],x};h[y]=en;
}
void dfs(int x)
{
for(int i=h[x];i;i=e[i].nx)if(e[i].t!=fa[x])
{
fa[e[i].t]=x;dep[e[i].t]=dep[x]+;dfs(e[i].t);
d[x]=max(d[x],d[e[i].t]+);
}
if(d[x]==K||x==rt)f[x]=x,d[x]=-;
}
void build(int x)
{
for(int i=h[x];i;i=e[i].nx)if(e[i].t!=fa[x])
{
if(!f[e[i].t])f[e[i].t]=f[x];
build(e[i].t);
}
}
void DFS(int x,int fa)
{
d[x]=++cnt;
for(int i=h[x];i;i=e[i].nx)
if(e[i].t!=fa)DFS(e[i].t,x);
}
bool CMP(const query&a,const query&b){return d[a.y]<d[b.y];}
void cal(int x)
{
if(u[x]^=)ans+=a[x]?A[b[x]]:B[b[x]],a[x]?++B[b[x]]:++A[b[x]];
else ans-=a[x]?A[b[x]]:B[b[x]],a[x]?--B[b[x]]:--A[b[x]];
}
void solve(int x,int fr)
{
d[x]=MN+;cal(x);
for(;rt==f[Q[nw].x]&&Q[nw].y==x;++nw)
{
for(int i=Q[nw].x,u=;i!=rt;i=fa[i])
if(d[i]==MN+&&u)cal(rt),u=;else cal(i);
C[Q[nw].id]=ans;
for(int i=Q[nw].x,u=;i!=rt;i=fa[i])
if(d[i]==MN+&&u)cal(rt),u=;else cal(i);
}
for(int i=h[x];i;i=e[i].nx)
if(e[i].t!=fr)solve(e[i].t,x);
d[x]=;cal(x);
}
int main()
{
BB[fread(BB,,<<,stdin)]=;
int n=read(),i,j,q;
for(i=;i<=n;++i)a[i]=read();
for(i=;i<=n;++i)b[i]=mp[b[i]=read()]?mp[b[i]]:mp[b[i]]=++cnt;
for(i=;i<n;++i)ins(read(),read());
rt=%n+;dfs(rt);build(rt);
for(q=read(),i=;i<=q;++i)
{
Q[i].x=read();Q[i].y=read();Q[i].id=i;
if(dep[Q[i].x]-dep[f[Q[i].x]]>dep[Q[i].y]-dep[f[Q[i].y]])swap(Q[i].x,Q[i].y);
}
sort(Q+,Q+q+,cmp);
for(nw=;nw<=q;)
{
DFS(f[Q[nw].x],cnt=);
for(i=nw;f[Q[i].x]==f[Q[nw].x];++i);
sort(Q+nw,Q+i,CMP);
solve(rt=f[Q[nw].x],ans=);
}
for(i=;i<=q;++i)printf("%I64d\n",C[i]);
}