求助UKE
查看原帖
求助UKE
849602
封禁用户楼主2025/1/30 21:13
#include<bits/stdc++.h>
using namespace std;
struct node{
	int l,r,ll,rr,id,lca;
}q[211111]; 
int n,m,a[211111],st[411111],ed[411111],dfn[211111],f[211111],num,size[211111],his[211111],dep[211111],son[211111],top[211111],c[211111],tmp,blo,l=1,r,use[211111],ans[211111],data[211111];
vector<int>d[211111];
void dfs1(int u,int fa){
	f[u]=fa,st[u]=++num,size[u]=1,his[num]=u,dep[u]=dep[fa]+1;
	vector<int>::iterator it;
	for (it=d[u].begin();it!=d[u].end();it++){
		int v=(*it);
		if(v==fa)continue;
		dfs1(v,u);
		size[u]+=size[v];
		if (size[v]>size[son[u]])son[u]=v;
	}
	ed[u]=++num,his[num]=u;
}
void dfs2(int u,int to){
	top[u]=to;
	if(son[u])dfs2(son[u],to);
	vector<int>::iterator it;
	for (it=d[u].begin();it!=d[u].end();it++){
		int v=(*it);
		if(v!=son[u]&&v!=f[u])dfs2(v,v);
	}
}
int Lca(int x,int y){
	while(top[x]!=top[y]){
		if(dep[top[x]]<dep[top[y]])swap(x,y);
		x=f[top[x]];
	}
	if(dep[x]>dep[y])swap(x,y);
	return x;
}
void add(int x){
	tmp+=(++c[a[x]]==1);
}
void del(int x){
	tmp-=(--c[a[x]])==0;
}
void calc(int x){
	(!use[x])?add(x):del(x);
	use[x]^=1;
}
int cmp(node x,node y){
	return (x.ll=y.ll)?(x.ll%2==1?x.r<y.r:x.r>y.r):x.l<y.l;
}
int main() {
	cin>>n>>m;
	for(int i=1;i<=n;i++)cin>>a[i],data[i]=a[i];
	sort(data+1,data+n+1);
	for(int i=1;i<=n;i++)a[i]=lower_bound(data+1,data+1+n,a[i])-data;
	for(int i=1,x,y;i<n;i++){
		cin>>x>>y;
		d[x].push_back(y);
		d[y].push_back(x);
	}
	dfs1(1,0);
	dfs2(1,1);
	blo=n*2/sqrt(m*2/3);
	for(int i=1,x,y;i<=m;i++){
		cin>>x>>y;
		if(st[x]>st[y])swap(x,y);
		q[i].id=i,q[i].lca=Lca(x,y);
		if(q[i].lca==x)q[i]={st[x],st[y],st[x]/blo,st[y]/blo,i,0};
		else q[i]={ed[x],st[y],ed[x]/blo,st[y]/blo,i,Lca(x,y)};
	}
	sort(q+1,q+m+1,cmp);
	for(int i=1;i<=m;i++){
		while(l>q[i].l)calc(his[--l]);
		while(r<q[i].r)calc(his[++r]);
		while(l<q[i].l)calc(his[l++]);
		while(r>q[i].r)calc(his[r--]);
		if(q[i].lca)calc(q[i].lca);
		ans[q[i].id]=tmp;
		if(q[i].lca)calc(q[i].lca);
	}
	for(int i=1;i<=m;i++)cout<<ans[i]<<endl;
	return 0;
}
2025/1/30 21:13
加载中...