P2597 [ZJOI2012]灾难（拓扑、LCA）

题目链接

题目简述

给你食物链中所有生物的食物（有向无环图DAG），现在问，如果每个生物i消失了会有多少生物也随之消失。

思路

• 我们很容易想到拓扑，确实纯拓扑也可以做（注意：最开始是单个生物入队，不是入度为零的入队），但是会超时。
• 我们可以深入思考一下，注意到如果要使一种生物灭亡，那么一定要他的食物都灭亡，而一种生物灭亡可能会导致很多生物都灭亡，这不就是一对一对多吗？也就是说我们可以构建一棵树，这棵树中的父亲节点灭亡了，那么他的儿子节点都会灭亡。所以结果就是某节点的子树大小。

暴力拓扑代码

#include<cstdio>
#include<queue>
#define ri register int
const int maxn=666660;
struct E{
	int v,n;
}e[maxn];
int head[maxn];
int in0[maxn],in[maxn];
int cnt,ans;
inline int read(){
	int x=0,f=0;char c=getchar();
	while(c>'9'||c<'0') f=c=='-',c=getchar();
	while(c>='0'&&c<='9') x=(x<<3)+(x<<1)+c-'0',c=getchar();
	return f?-x:x;
}
inline void add(int u,int v){
	e[++cnt].v=v;
	e[cnt].n=head[u];
	head[u]=cnt;
}
void topsort(int x){
	std::queue<int>q;
	q.push(x);//只有最开始灭绝的生物入队 
	while(q.size())
	{
		int u=q.front();q.pop();
		for(ri i=head[u];i;i=e[i].n)
		{
			int v=e[i].v;
			if(i%2==0) continue;
			in[v]--;
			if(in[v]==0) ans++,q.push(v);//统计答案 
		}
	}
}
int main(){
	int n=read();
	for(ri i=1;i<=n;++i) 
	{
		int a=read();
		while(a) add(a,i),add(i,a),in0[i]++,a=read();
	}
	for(ri i=1;i<=n;++i)
	{
		ans=0;
		for(ri j=1;j<=n;++j) in[j]=in0[j];//每次in会变，所以要处理一下 
		topsort(i);
		printf("%d\n",ans);
	}
	return 0;
}

AC代码

#include<cstdio>
#include<queue>
#define ri register int
const int maxn=66666;
const int maxm=2e6+9;
struct E{
	int v,n;
}e[maxm];
int head[maxn];
int in[maxn];
int dad[maxn];
struct E1{
	int v,n;
}e1[maxm];
int head1[maxn];
int d[maxn],lg[maxn],fa[maxn][19];
int cnt,cnt1,n;
inline int read(){
	int x=0,f=0;char c=getchar();
	while(c>'9'||c<'0') f=c=='-',c=getchar();
	while(c>='0'&&c<='9') x=(x<<3)+(x<<1)+c-'0',c=getchar();
	return f?-x:x;
}
inline void add(int u,int v){
	e[++cnt].v=v;
	e[cnt].n=head[u];
	head[u]=cnt;
}
inline void add1(int u,int v){
	e1[++cnt1].v=v;
	e1[cnt1].n=head1[u];
	head1[u]=cnt1;
}
int lca(int x,int y){
	if(d[x]<d[y]) return lca(y,x);
	while(d[x]>d[y]) x=fa[x][lg[d[x]-d[y]]];
	if(x==y) return x;
	for(ri i=lg[d[x]];i>=0;--i)
		if(fa[x][i]!=fa[y][i])
			x=fa[x][i],y=fa[y][i];
	return fa[x][0];
}
void topsort(){//拓扑 
	std::queue<int>q;
	for(ri i=1;i<=n;++i) if(!in[i]) dad[i]=0,q.push(i);//入队 
	while(q.size())
	{
		int u=q.front();q.pop();
		add1(dad[u],u),d[u]=d[dad[u]]+1,fa[u][0]=dad[u];//求出新树的数据 
		for(ri i=1;i<=lg[d[u]]+1;++i) fa[u][i]=fa[fa[u][i-1]][i-1];
		for(ri i=head[u];i;i=e[i].n)
		{
			int v=e[i].v;
			if(dad[v]==-1) dad[v]=u;
			else dad[v]=lca(dad[v],u);//求lca作为父亲 
			in[v]--;
			if(!in[v]) q.push(v);
		}
	}
}
int s[maxn];
void dfs(int u){//计算子树大小 
	s[u]=1;
	for(ri i=head1[u];i;i=e1[i].n)
	{
		int v=e1[i].v;
		dfs(v);
		s[u]+=s[v];
	}
}
int main(){
	n=read();
	for(ri i=1;i<=n;++i) 
	{
		dad[i]=-1,lg[i]=lg[i-1]+(1<<lg[i-1]==i);
		int a=read();
		while(a) add(a,i),in[i]++,a=read();
	}
	for(ri i=1;i<=n;++i) lg[i]--;//预处理 
	topsort();
	dfs(0);//从0开始搜 
	for(ri i=1;i<=n;++i) printf("%d\n",s[i]-1);//记得减一 
	return 0;
}