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;
}