1. 程式人生 > >BZOJ3757 蘋果樹

BZOJ3757 蘋果樹

tin print scan top namespace str operator swap const

一道挺好的樹上莫隊板子題,可惜存在版權無法提交了。

這裏給出代碼供大家參考,分塊操作類似於王室聯盟,轉成序列後和普通莫隊一樣。

By:大奕哥

  1 #include<bits/stdc++.h>
  2 using namespace std;
  3 const int N=1e5+5;
  4 int d[N],fa[N][20],bel[N],size[N],dfn[N],col[N],ans[N],p[N],head[N];
  5 int n,m,sum,idx,cnt,B,num,top,s[N];
  6 bool v[N];
  7 struct
node{ 8 int to,nex; 9 }e[N<<1]; 10 void add(int x,int y) 11 { 12 e[++cnt].nex=head[x];e[cnt].to=y;head[x]=cnt; 13 } 14 struct quer{ 15 int x,y,a,b,id; 16 bool operator<(const quer &b)const{ 17 return bel[x]<bel[b.x]||(bel[x]==bel[b.x]&&dfn[y]<dfn[b.y]);
18 } 19 }q[N]; 20 void dfs(int x) 21 { 22 dfn[x]=++idx;s[++top]=x; 23 for(int i=1;i<=16;++i) 24 fa[x][i]=fa[fa[x][i-1]][i-1]; 25 for(int i=head[x];i;i=e[i].nex) 26 { 27 int y=e[i].to; 28 if(y==fa[x][0])continue; 29 fa[y][0]=x;d[y]=d[x]+1
; 30 dfs(y); 31 size[x]+=size[y]; 32 if(size[x]>=B) 33 { 34 num++; 35 for(int k=1;k<=size[x];++k) 36 bel[s[top--]]=num; 37 size[x]=0; 38 } 39 } 40 size[x]++; 41 } 42 int lca(int x,int y) 43 { 44 if(d[x]<d[y])swap(x,y); 45 int tmp=d[x]-d[y]; 46 for(int i=0;i<=16;++i) 47 if(tmp&(1<<i))x=fa[x][i]; 48 for(int i=16;i>=0;--i) 49 { 50 if(fa[x][i]!=fa[y][i]) 51 { 52 x=fa[x][i]; 53 y=fa[y][i]; 54 } 55 } 56 return x==y?x:fa[x][0]; 57 } 58 void reverse(int x) 59 { 60 if(!v[x]){v[x]=1;p[col[x]]++;if(p[col[x]]==1)sum++;} 61 else {v[x]=0;p[col[x]]--;if(p[col[x]]==0)sum--;} 62 return; 63 } 64 void solve(int x,int y) 65 { 66 while(x!=y) 67 { 68 if(d[x]<d[y])swap(x,y); 69 reverse(x);x=fa[x][0]; 70 } 71 return; 72 } 73 int main() 74 { 75 scanf("%d%d",&n,&m); 76 B=sqrt(n);int x,y,rt; 77 for(int i=1;i<=n;++i)scanf("%d",&col[i]); 78 for(int i=1;i<=n;++i) 79 { 80 scanf("%d%d",&x,&y); 81 if(!x)rt=y;else if(!y)rt=x; 82 else add(x,y),add(y,x); 83 } 84 dfs(rt);num++; 85 while(top)bel[s[top--]]=num; 86 for(int i=1;i<=m;++i) 87 { 88 scanf("%d%d",&q[i].x,&q[i].y); 89 if(dfn[q[i].x]>dfn[q[i].y])swap(q[i].x,q[i].y); 90 q[i].id=i; 91 scanf("%d%d",&q[i].a,&q[i].b); 92 } 93 sort(q+1,q+1+m); 94 int z=lca(q[1].x,q[1].y); 95 solve(q[1].x,q[1].y); 96 reverse(z);ans[q[1].id]=sum; 97 if(p[q[1].a]&&p[q[1].b]&&q[1].a!=q[1].b)ans[q[1].id]--; 98 reverse(z); 99 for(int i=2;i<=m;++i) 100 { 101 int z=lca(q[i].x,q[i].y); 102 solve(q[i-1].x,q[i].x); 103 solve(q[i-1].y,q[i].y); 104 reverse(z);ans[q[i].id]=sum; 105 if(p[q[i].a]&&p[q[i].b]&&q[i].a!=q[i].b)ans[q[i].id]--; 106 reverse(z); 107 } 108 for(int i=1;i<=m;++i)printf("%d\n",ans[i]); 109 return 0; 110 }

BZOJ3757 蘋果樹