題意:
思路:
樹上分塊方法<-由這個題,我們可以得出一種分塊方法,他能保證每個塊中內部移動次數,同時也能保證塊的大小
這個是講述怎麼轉移的因爲轉移的時候,lca非常麻煩,要各種分類討論,這個博客告訴我們,只要不管lca進行轉移,最後計算結果的時候臨時加上lca即可
錯誤及反思:
第一個樹上莫隊,所以跑得有點慢,寫的也有點長,以後寫的好了再改改
代碼:
#include<bits/stdc++.h>
using namespace std;
const int N = 101000;
map<int,int> mp;
vector<int> son[N];
stack<int> s;
int blocks,block[N],cnt[N],ans[N],to[N][20],tot=0,tid=0,depth[N],now=0,n,m,val[N];
bool vis[N];
struct Q{
int l,r,id;
}q[N];
bool cmp(Q a,Q b){
if(block[a.l]!=block[b.l])
return block[a.l]<block[b.l];
return block[a.r]<block[b.r];
}
void dfs(int now,int fa,int dep){
int si=s.size();
to[now][0]=fa;
depth[now]=dep;
for(int i=0;i<son[now].size();i++){
if(son[now][i]!=fa){
dfs(son[now][i],now,dep+1);
if(s.size()-si>=blocks){
while(s.size()!=si){
block[s.top()]=tot;
s.pop();
}
tot++;
}
}
}
s.push(now);
}
void getlca(){
for(int i=1;i<=18;i++)
for(int j=1;j<=n;j++)
to[j][i]=to[to[j][i-1]][i-1];
}
int lca(int a,int b){
if(depth[a]>depth[b])
swap(a,b);
for(int i=18;i>=0;i--)
if(depth[to[b][i]]>=depth[a])
b=to[b][i];
if(a==b) return a;
for(int i=18;i>=0;i--){
if(to[a][i]!=to[b][i]){
a=to[a][i];
b=to[b][i];
}
}
return to[a][0];
}
void change(int u,int v){
int zx=lca(u,v);
while(u!=zx){
if(vis[u]){
cnt[val[u]]--;
if(cnt[val[u]]==0)
now--;
}
else{
cnt[val[u]]++;
if(cnt[val[u]]==1)
now++;
}
vis[u]=!vis[u];
u=to[u][0];
}
while(v!=zx){
if(vis[v]){
cnt[val[v]]--;
if(cnt[val[v]]==0)
now--;
}
else{
cnt[val[v]]++;
if(cnt[val[v]]==1)
now++;
}
vis[v]=!vis[v];
v=to[v][0];
}
}
int main(){
scanf("%d%d",&n,&m);
blocks=sqrt(n);
for(int i=1;i<=n;i++){
scanf("%d",&val[i]);
if(!mp[val[i]]) mp[val[i]]=val[i]=++tid;
else val[i]=mp[val[i]];
}
for(int i=1;i<n;i++){
int ta,tb;
scanf("%d%d",&ta,&tb);
son[ta].push_back(tb);
son[tb].push_back(ta);
}
for(int i=1;i<=m;i++){
scanf("%d%d",&q[i].l,&q[i].r);
q[i].id=i;
}
dfs(1,1,1);
while(s.empty()){
block[s.top()]=tot-1;
s.pop();
}
getlca();
sort(q+1,q+m+1,cmp);
for(int i=1,l=1,r=1;i<=m;i++){
if(l!=q[i].l) change(l,q[i].l);
if(r!=q[i].r) change(r,q[i].r);
int zx=lca(q[i].l,q[i].r);
if(cnt[val[zx]]==0)
ans[q[i].id]=now+1;
else ans[q[i].id]=now;
l=q[i].l;
r=q[i].r;
}
for(int i=1;i<=m;i++)
printf("%d\n",ans[i]);
}