【SPOJ】Count On A Tree II（树上莫队）
题面
洛谷上有翻译啦
题解
如果不在树上就是一个很裸很裸的莫队
现在在树上，就是一个很裸很裸的树上莫队啦。
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define ll long long
#define RG register
#define MAX 111111
inline int read()
{
RG int x=0,t=1;RG char ch=getchar();
while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
if(ch=='-')t=-1,ch=getchar();
while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
return x*t;
}
struct Line{int v,next;}e[MAX];
int h[MAX],cnt=1;
inline void Add(int u,int v){e[cnt]=(Line){v,h[u]};h[u]=cnt++;}
int S[MAX],a[MAX],top,G[MAX],gr;
int f[16][MAX],dep[MAX],dfn[MAX],tim;
int n,m,blk,ans,Ans[MAX],sum[MAX];
bool vis[MAX];
int dfs(int u,int ff)
{
f[0][u]=ff;dep[u]=dep[ff]+1;dfn[u]=++tim;
for(int i=1;i<=15;++i)f[i][u]=f[i-1][f[i-1][u]];
int ret=0;
for(int i=h[u];i;i=e[i].next)
{
int v=e[i].v;if(v==ff)continue;
ret+=dfs(v,u);
if(ret>=blk){++gr;while(ret--)G[S[top--]]=gr;ret=0;}
}
S[++top]=u;
return ret+1;
}
int LCA(int u,int v)
{
if(dep[u]<dep[v])swap(u,v);
for(int i=15;~i;--i)if(dep[f[i][u]]>=dep[v])u=f[i][u];
if(u==v)return u;
for(int i=15;~i;--i)if(f[i][u]!=f[i][v])u=f[i][u],v=f[i][v];
return f[0][u];
}
struct query{int id,u,v,lb,rb;}q[MAX];
bool operator<(query a,query b){if(a.lb!=b.lb)return a.lb<b.lb;return a.rb<b.rb;}
void Change(int x)
{
if(vis[x])--sum[a[x]],ans-=!sum[a[x]];
else ans+=!sum[a[x]],++sum[a[x]];
vis[x]^=1;
}
void ModifyLink(int u,int v)
{
while(u!=v)
{
if(dep[u]<dep[v])swap(u,v);
Change(u);u=f[0][u];
}
}
int main()
{
n=read();m=read();blk=sqrt(n);
for(int i=1;i<=n;++i)S[i]=a[i]=read();
sort(&S[1],&S[n+1]);top=unique(&S[1],&S[n+1])-S-1;
for(int i=1;i<=n;++i)a[i]=lower_bound(&S[1],&S[top+1],a[i])-S;
for(int i=1,u,v;i<n;++i)u=read(),v=read(),Add(u,v),Add(v,u);
top=0;dfs(1,0);
for(int i=1;i<=m;++i)
{
int u=read(),v=read();if(dfn[u]>dfn[v])swap(u,v);
q[i]=(query){i,u,v,G[u],G[v]};
}
sort(&q[1],&q[m+1]);
int lca=LCA(q[1].u,q[1].v);
ModifyLink(q[1].u,q[1].v);
Change(lca);Ans[q[1].id]=ans;Change(lca);
for(int i=2;i<=m;++i)
{
ModifyLink(q[i-1].u,q[i].u);
ModifyLink(q[i-1].v,q[i].v);
lca=LCA(q[i].u,q[i].v);
Change(lca);Ans[q[i].id]=ans;Change(lca);
}
for(int i=1;i<=m;++i)printf("%d
",Ans[i]);
return 0;
}