题意
给定n个节点的一棵树,每个节点上都有一个数字。
m次询问,没次求u到v的路径上有几种不同的数字。
题解
树上莫队经典题。
首先要会树分块,见bzoj1086: [SCOI2005]王室联邦。
与普通莫队类似,记dfn[x]为x的入栈时间戳,我们按(belong[U], dfn[V])排序所有询问。
显然分块完后属于同一块的最远点对距离不超过块的大小,
而且考虑括号序列,dfn[V]递增的话一步一步走过去不会超过2n部。
因此复杂度和普通莫队类似。
如何从当前状态转移到下一个呢?可以发现对于LCA那个点有点特殊。
对于这题,如果我们得到除了lca那个点其他的信息,只需要加上lca那一个点即可得到答案。
设S(u,v)为从u到v的路径除去LCA那个点的信息,T(x)表示根到x的信息。
如果我们把v改为v2 ,考虑变化了哪些(下面的加是指类似异或的运算,即有2个相同元素就约掉):
S(u,v)+S(u,v2)
=(T(u)+T(v)+(T(u)+T(v2))
=(T(v)+T(v2))=S(v,v2)
也就是说,只要把v到v2的路径上除了lca(v,v2)的点的存在性取反即可。
复杂度
O(nn−−√)
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
const int maxn=40005, maxe=80005, maxq=100005;
int n,m,c[maxn],b[maxn],sum[maxn],ans,dfn[maxn],Tim,G,blk,blg[maxn],stk[maxn],top,id_c[maxn];
int son[maxe],fir[maxn],nxt[maxe],tot;
bool vis[maxn];
int dfs(int x,int pre){
dfn[x]=++Tim; int bottom=top;
for(int j=fir[x];j;j=nxt[j]) if(son[j]!=pre){
dfs(son[j],x);
if(top-bottom>=blk){
G++;
while(top!=bottom) blg[stk[top--]]=G;
}
}
stk[++top]=x;
}
struct data{ int x,y,id,res; } q[maxq];
bool my_cmp1(const data &A,const data &B){
if(blg[A.x]==blg[B.x]) return dfn[A.y]<dfn[B.y];
return blg[A.x]<blg[B.x];
}
bool my_cmp2(const data &A,const data &B){ return A.id<B.id; }
void add(int x,int y){
son[++tot]=y; nxt[tot]=fir[x]; fir[x]=tot;
}
int anc[maxn][20],dep[maxn];
void dfs_LCA(int x,int pre){
anc[x][0]=pre; for(int i=1;i<=17;i++) anc[x][i]=anc[anc[x][i-1]][i-1];
for(int j=fir[x];j;j=nxt[j]) if(son[j]!=pre) dep[son[j]]=dep[x]+1, dfs_LCA(son[j],x);
}
int LCA(int x,int y){
if(dep[x]<dep[y]) swap(x,y);
for(int j=17;j>=0;j--) if(dep[anc[x][j]]>=dep[y]) x=anc[x][j];
if(x==y) return x;
for(int j=17;j>=0;j--) if(anc[x][j]!=anc[y][j]) x=anc[x][j], y=anc[y][j];
return anc[x][0];
}
int find(int x){
int L=1,R=b[0];
while(L<=R){
int mid=(L+R)>>1;
if(b[mid]==x) return mid;
if(b[mid]<x) L=mid+1; else R=mid-1;
}
}
void rev(int x){
if(vis[x]) ans=(--sum[id_c[x]])?ans:ans-1;
else ans=(sum[id_c[x]]++)?ans:ans+1;
vis[x]^=1;
}
void work(int x,int y){
while(x!=y){
if(dep[x]<dep[y]) swap(x,y);
rev(x); x=anc[x][0];
}
}
int main(){
freopen("cot2.in","r",stdin);
freopen("cot2.out","w",stdout);
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) scanf("%d",&c[i]), b[++b[0]]=c[i];
for(int i=1;i<=n-1;i++){
int x,y; scanf("%d%d",&x,&y);
add(x,y); add(y,x);
}
dfs_LCA(1,1);
blk=sqrt(n); dfs(1,0);
for(int i=1;i<=m;i++){
scanf("%d%d",&q[i].x,&q[i].y); q[i].id=i;
if(dfn[q[i].x]>dfn[q[i].y]) swap(q[i].x,q[i].y);
}
sort(b+1,b+1+b[0]); b[0]=unique(b+1,b+1+b[0])-(b+1);
sort(q+1,q+1+m,my_cmp1);
for(int i=1;i<=n;i++) id_c[i]=find(c[i]);
q[0].x=q[0].y=1;
for(int i=1;i<=m;i++){
work(q[i-1].x,q[i].x);
work(q[i-1].y,q[i].y);
int _lca=LCA(q[i].x,q[i].y);
rev(_lca); q[i].res=ans; rev(_lca);
}
sort(q+1,q+1+m,my_cmp2);
for(int i=1;i<=m;i++) printf("%d\n",q[i].res);
return 0;
}