首页 > 代码库 > AC日记——Count on a tree II spoj
AC日记——Count on a tree II spoj
Count on a tree II
思路:
树上莫队;
先分块,然后,就好办了;
来,上代码:
#include <cmath>#include <cstdio>#include <cstring>#include <iostream>#include <algorithm>using namespace std;#define maxn 40005#define maxm 100005struct 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
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。