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;
}