SP10707 Count on a tree II

Description

给定一棵N个节点的树，每个点有一个权值，对于M个询问(u,v)，你需要回答u和v这两个节点间有多少种不同的点权。

Input

第一行两个整数N,M。
第二行有N个整数，其中第i个整数表示点i的权值。
后面N-1行每行两个整数(x,y)，表示点x到点y有一条边。
最后M行每行两个整数(u,v)，表示一组询问。

数据范围是N<=40000 M<=100000 点权在int范围内

Output

M行，表示每个询问的答案。

Sample Input

8 2
105 2 9 3 8 5 7 7
1 2
1 3
1 4
3 5
3 6
3 7
4 8
2 5
3 8

Sample Output

4

4

树上莫队模板。（强制在线的话可以树分块）。只是提一句拉成dfs序之后并不是简单的增加删减，而是走过两次的会被删除，只出现一次的才真的加进去，切记切记。（我权值不离散化套个log=tle）

#include<bits/stdc++.h>
using namespace std;
const int Maxn=100005;
inline int getint(){
	int res=0;char c=getchar();
	while(!isdigit(c))c=getchar();
	while(isdigit(c))res=res*10+c-'0',c=getchar();
	return res;
}
vector<int>G[Maxn];
int n,m,siz,v[Maxn],bl[Maxn];
int cnt,p[Maxn],fir[Maxn],sec[Maxn];
struct Query{
	int l,r,lca,idx;
}q[Maxn];
bool cmp(const Query&A,const Query&B){
	if(bl[A.l]!=bl[B.l])return A.l<B.l;
	return bl[A.l]&1?A.r<B.r:A.r>B.r;
}
struct Tree{
	int dep[Maxn],prt[20][Maxn];
	void dfs(int x,int fa,int Dep){
		p[fir[x]=++cnt]=x;
		prt[0][x]=fa;dep[x]=Dep;
		for(int i=0;i<G[x].size();++i)
			if(G[x][i]!=fa)dfs(G[x][i],x,Dep+1);
		p[sec[x]=++cnt]=x;
	}
	void ST(){
		int maxlog=log2(n);
		for(int j=1;j<=maxlog;++j)
			for(int i=1;i<=n;++i)
				if(prt[j-1][i])prt[j][i]=prt[j-1][prt[j-1][i]];
	}
	int lca(int x,int y){
		if(dep[x]<dep[y])swap(x,y);
		for(int i=log2(dep[x]);i>=0;--i)
			if(dep[x]-(1<<i)>=dep[y])x=prt[i][x];
		if(x==y)return y;
		for(int i=log2(dep[x]);i>=0;--i)
			if(prt[i][x]!=prt[i][y])x=prt[i][x],y=prt[i][y];
		return prt[0][x];
	}
}st;
int ret,ans[Maxn],Count[Maxn];
bool vst[Maxn];
void update(int x){
	vst[x=p[x]]^=1;
	if(vst[x]){
		if(++Count[v[x]]==1)++ret;
	}else {
		if(--Count[v[x]]==0)--ret;
	}
}
int tmp[Maxn];
void discrete(){
	for(int i=1;i<=n;++i)tmp[i]=v[i];
	sort(tmp+1,tmp+n+1);
	int siz=unique(tmp+1,tmp+n+1)-tmp-1;
	for(int i=1;i<=n;++i)
		v[i]=lower_bound(tmp+1,tmp+siz+1,v[i])-tmp;
}
int main(){
	n=getint(),m=getint();
	for(int i=1;i<=n;++i)
		v[i]=getint();
	discrete(); 
	for(int i=1;i<n;++i){
		int x=getint(),y=getint();
		G[x].push_back(y);
		G[y].push_back(x);
	}
	st.dfs(1,0,0);
	st.ST();
	for(int i=1;i<=m;++i){
		int x=getint(),y=getint(),lca=st.lca(x,y);
		if(fir[x]>fir[y])swap(x,y);
		if(lca==x){
			q[i]=(Query){fir[x],fir[y],0,i};
		}
		else {
			q[i]=(Query){sec[x],fir[y],lca,i};
		}
	}
	siz=sqrt(cnt);
	for(int i=1;i<=cnt;++i)
		bl[i]=(i-1)/siz+1;
	sort(q+1,q+m+1,cmp);
	int l=1,r=0;
	for(int i=1;i<=m;++i){
		while(l<q[i].l)update(l++);
		while(l>q[i].l)update(--l);
		while(r>q[i].r)update(r--);
		while(r<q[i].r)update(++r);
		ans[q[i].idx]=ret+(q[i].lca&&!Count[v[q[i].lca]]);
		
	}
	for(int i=1;i<=m;++i)
		cout<<ans[i]<<"\n";
	return 0;
}