bzoj 2815: [ZJOI2012]灾难|倍增lca|思路题

详见fhq blog

主要是先拓扑..然后构建新树 然后倍增求儿子

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int MAXN=700000;
struct EDGE
{
int tot,g[MAXN],nnext[MAXN],num[MAXN];
void add(int x,int y){tot++;nnext[tot]=g[x];g[x]=tot;num[tot]=y;}
}fa,son,ans;
int n,size[MAXN],in[MAXN],to[MAXN][18],depth[MAXN],team[MAXN],head,tail;
int lca(int x,int y)
{
if(depth[x]<depth[y]) swap(x,y);
for(int i=17;i>=0;i--) if(depth[to[x][i]]>=depth[y]) x=to[x][i];
if(x==y) return x;
for(int i=17;i>=0;i--) if(to[x][i]!=to[y][i]) x=to[x][i],y=to[y][i];
return to[x][0];
}
void bfs()
{
team[++tail]=n+1;
while(head<tail)
{
int x=team[++head];
int root=fa.num[fa.g[x]];
for(int i=fa.g[x];i;i=fa.nnext[i])
{
int tmp=fa.num[i];
root=lca(root,tmp);
}
depth[x]=depth[root]+1;
to[x][0]=root;
ans.add(root,x);
for(int i=1;i<=17;i++) to[x][i]=to[to[x][i-1]][i-1];

for(int i=son.g[x];i;i=son.nnext[i])
{
int tmp=son.num[i];
in[tmp]--;
if(in[tmp]==0)
team[++tail]=tmp;
}
}
}
void dfs(int x)
{
size[x]=1;
for(int i=ans.g[x];i;i=ans.nnext[i])
{
int tmp=ans.num[i];
dfs(tmp);
size[x]+=size[tmp];
}
}
int main()
{
scanf("%d",&n);
for(int x,i=1;i<=n;i++) while(scanf("%d",&x)==1&&x!=0) son.add(x,i),in[i]++,fa.add(i,x);
for(int i=1;i<=n;i++) if(in[i]==0) son.add(n+1,i),in[i]++,fa.add(i,n+1);
bfs();
dfs(n+1);
for(int i=1;i<=n;i++) printf("%d\n",size[i]-1);
return 0;
}