AC日记——Count on a tree II spoj
Posted Only U - IU
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了AC日记——Count on a tree II spoj相关的知识,希望对你有一定的参考价值。
思路:
树上莫队;
先分块,然后,就好办了;
来,上代码:
#include <cmath> #include <cstdio> #include <cstring> #include <iostream> #include <algorithm> using namespace std; #define maxn 40005 #define maxm 100005 struct QueryType { int u,v,id; }; struct QueryType qu[maxm]; int n,m,ti[maxn],num[maxn],Hash[maxn],siz; int bel[maxn],f[maxn],deep[maxn],ans[maxm]; int E[maxn<<1],V[maxn<<1],head[maxn],cnt=0; int top[maxn],size[maxn],dis[maxn],sizee; bool if_[maxn]; inline void in(int &now) { char Cget=getchar();now=0; while(Cget>‘9‘||Cget<‘0‘) Cget=getchar(); while(Cget>=‘0‘&&Cget<=‘9‘) { now=now*10+Cget-‘0‘; Cget=getchar(); } } void pre(int now,int fa) { f[now]=fa,deep[now]=deep[fa]+1; bel[now]=(++cnt+1)/siz,size[now]=1; for(int i=head[now];i;i=E[i]) { if(V[i]==fa) continue; pre(V[i],now),size[now]+=size[V[i]]; } } void dfs(int now,int chain) { top[now]=chain;int pos=0; for(int i=head[now];i;i=E[i]) { if(V[i]==f[now]) continue; if(size[V[i]]>size[pos]) pos=V[i]; } if(pos==0) return ; dfs(pos,chain); for(int i=head[now];i;i=E[i]) { if(V[i]==f[now]||V[i]==pos) continue; dfs(V[i],V[i]); } } int solve_lca(int x,int y) { while(top[x]!=top[y]) { if(deep[top[x]]<deep[top[y]]) swap(x,y); x=f[top[x]]; } if(deep[x]>deep[y]) swap(x,y); return x; } bool cmp(QueryType aa,QueryType bb) { if(bel[aa.u]==bel[bb.u]) return bel[aa.v]<bel[bb.v]; else return bel[aa.u]<bel[bb.u]; } inline void updata(int to) { if(if_[to]) { ti[num[to]]--; if(ti[num[to]]==0) cnt--; } else { ti[num[to]]++; if(ti[num[to]]==1) cnt++; } if_[to]=!if_[to]; } int main() { in(n),in(m);siz=sqrt(n); for(int i=1;i<=n;i++) in(num[i]),Hash[i]=num[i]; sort(Hash+1,Hash+n+1); sizee=unique(Hash+1,Hash+n+1)-Hash-1;int u,v; for(int i=1;i<=n;i++) num[i]=lower_bound(Hash+1,Hash+sizee+1,num[i])-Hash; for(int i=1;i<n;i++) { in(u),in(v); E[++cnt]=head[u],V[cnt]=v,head[u]=cnt; E[++cnt]=head[v],V[cnt]=u,head[v]=cnt; } cnt=0,pre(1,0),dfs(1,1); for(int i=1;i<=m;i++) { in(qu[i].u),in(qu[i].v),qu[i].id=i; if(bel[qu[i].u]>bel[qu[i].v]) swap(qu[i].u,qu[i].v); } sort(qu+1,qu+m+1,cmp);u=1,v=1,cnt=0; for(int no=1;no<=m;no++) { int lca=solve_lca(u,qu[no].u); while(u!=f[lca]) updata(u),u=f[u];u=qu[no].u; while(u!=f[lca]) updata(u),u=f[u];u=qu[no].u; lca=solve_lca(v,qu[no].v); while(v!=f[lca]) updata(v),v=f[v];v=qu[no].v; while(v!=f[lca]) updata(v),v=f[v];v=qu[no].v; lca=solve_lca(u,v); updata(lca),ans[qu[no].id]=cnt,updata(lca); } for(int i=1;i<=m;i++) printf("%d\n",ans[i]); return 0; }
以上是关于AC日记——Count on a tree II spoj的主要内容,如果未能解决你的问题,请参考以下文章
SPOJ.COT2 Count on a tree II(树上莫队)
spoj COT2 - Count on a tree II