P2597 [ZJOI2012]灾难（倍增LCA+拓扑排序）

据大佬说这玩意儿好像叫灾难树还是灭绝树？

我们先按建图，设点$u$的食物有$x[1]...x[k]$，即在图中这些点都有一条指向它的边

以样例来说，对于人，羊和牛都有一条指向它的边，然而不管是羊的灭绝还是牛的灭绝都不会导致人的灭绝，会导致人灭绝的只有草，也就是羊和牛的LCA

（严格来说这个应该不叫LCA，不过反正这是一个有向无环图，大家能理解就好）

于是我们把人挂到草的下面，表示只有草的灭绝会使人灭绝。

于是我们按拓扑序把每一个生物都依次加入灭绝树，然后一遍dfs，每一个生物的灾难值就是它的子树的大小减一

 //minamoto

 #include<iostream>

 #include<cstdio>

 using namespace std;

 #define getc() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++)

 char buf[<<],*p1=buf,*p2=buf;

 int read(){

     int res,f=;char ch;

     while((ch=getc())>''||ch<'')(ch=='-')&&(f=-);

     for(res=ch-'';(ch=getc())>=''&&ch<='';res=res*+ch-'');

     return res;

 }char sr[<<],z[];int C=-,Z=;

 inline void Ot(){fwrite(sr,,C+,stdout),C=-;}

 void print(int x){

     if(C><<)Ot();if(x<)sr[++C]='-',x=-x;

     while(z[++Z]=x%+,x/=);

     while(sr[++C]=z[Z],--Z);sr[++C]='\n';

 }

 const int N=;

 int head[N<<],Next[N<<],ver[N<<],tot,head1[N<<],ver1[N<<],Next1[N<<],tot1,head2[N<<],ver2[N<<],Next2[N<<],tot2;

 inline void add(int u,int v){ver[++tot]=v,Next[tot]=head[u],head[u]=tot;}

 inline void add1(int u,int v){ver1[++tot1]=v,Next1[tot1]=head1[u],head1[u]=tot1;}

 inline void add2(int u,int v){ver2[++tot2]=v,Next2[tot2]=head2[u],head2[u]=tot2;}

 int n,du[N],q[N],dep[N],fa[N][],ans[N];

 void topo(){

     int h=,t=;

     for(int i=;i<=n;++i)if(!du[i])q[++t]=i;

     while(h<=t){

         int u=q[h++];

         for(int i=head[u];i;i=Next[i]){

             int v=ver[i];

             if(--du[v]==)q[++t]=v;

         }

     }

 }

 int LCA(int u,int v){

     if(dep[u]<dep[v])swap(u,v);int k=dep[u]-dep[v];

     for(int i=;i<;++i)

     if((k>>i)&)u=fa[u][i];

     if(u==v)return u;

     for(int i=;i>=;--i)

     if(fa[u][i]!=fa[v][i])u=fa[u][i],v=fa[v][i];

     return fa[u][];

 }

 void dfs(int u){

     for(int i=head2[u];i;i=Next2[i])

     dfs(ver2[i]),ans[u]+=ans[ver2[i]];

     ++ans[u];

 }

 int main(){

 //    freopen("testdata.in","r",stdin);

     n=read();

     for(int i=,u;i<=n;++i){

         while(true){

             if((u=read())==)break;

             ++du[i],add1(i,u),add(u,i);

         }

     }

     topo();

     for(int i=;i<=n;++i){

         int u=ver1[head1[q[i]]];

         for(int j=head1[q[i]];j;j=Next1[j])u=LCA(u,ver1[j]);

         add2(u,q[i]),dep[q[i]]=dep[u]+,fa[q[i]][]=u;

         for(int j=;j<;++j)fa[q[i]][j]=fa[fa[q[i]][j-]][j-];

     }

     dfs();

     for(int i=;i<=n;++i)print(ans[i]-);

     return Ot(),;

 }

巴特西

P2597 [ZJOI2012]灾难（倍增LCA+拓扑排序）

最新文章

热门文章