当前位置: 代码迷 >> 综合 >> SPOJ COT2 Count on a tree II (树上莫队)
  详细解决方案

SPOJ COT2 Count on a tree II (树上莫队)

热度:31   发布时间:2023-12-17 03:28:12.0

题意:

思路:

     树上分块方法<-由这个题,我们可以得出一种分块方法,他能保证每个块中内部移动次数,同时也能保证块的大小
     这个是讲述怎么转移的因为转移的时候,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]);
}
  相关解决方案