[Codeforces]852I - Dating

题目大意：给定一棵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]);
}