【拓扑序+lca】P2597 [ZJOI2012]灾难

题目背景

阿米巴是小强的好朋友。

阿米巴和小强在草原上捉蚂蚱。小强突然想，如果蚂蚱被他们捉灭绝了，那么吃蚂蚱的小鸟就会饿死，而捕食小鸟的猛禽也会跟着灭绝，从而引发一系列的生态灾难。

学过生物的阿米巴告诉小强，草原是一个极其稳定的生态系统。如果蚂蚱灭绝了，小鸟照样可以吃别的虫子，所以一个物种的灭绝并不一定会引发重大的灾难。

题目描述

我们现在从专业一点的角度来看这个问题。我们用一种叫做食物网的有向图来描述生物之间的关系：

• 一个食物网有 n 个点，代表 n 种生物，生物从 1 到 n 编号。
• 如果生物 x 可以吃生物 y，那么从 y 向x 连一个有向边。
• 这个图没有环。
• 图中有一些点没有连出边，这些点代表的生物都是生产者，可以通过光合作用来生存。
• 而有连出边的点代表的都是消费者，它们必须通过吃其他生物来生存。
• 如果某个消费者的所有食物都灭绝了，它会跟着灭绝。

我们定义一个生物在食物网中的“灾难值”为，如果它突然灭绝，那么会跟着一起灭绝的生物的种数。

举个例子：在一个草场上，生物之间的关系如下

如果小强和阿米巴把草原上所有的羊都给吓死了，那么狼会因为没有食物而灭绝，而小强和阿米巴可以通过吃牛、牛可以通过吃草来生存下去。所以，羊的灾难值是 1。但是，如果草突然灭绝，那么整个草原上的 5 种生物都无法幸免，所以，草的灾难值是 4。

给定一个食物网，你要求出每个生物的灾难值。

输入格式

第一行有一个整数，表示食物网的结点个数 n。

第 2 到第 (n+1) 行，每行若干个互不相同的整数，第 (i+1) 行的整数 ai,j​ 表示编号为 i 的生物可以吃编号为 ai​ 的生物。每行结尾有一个整数0 表示本行结束。

输出格式

输出 n行，每行一个整数，第 i 行输出编号为 i的生物的灾难值。

输入输出样例

输入 #1

5
0
1 0
1 0
2 3 0
2 0

输出 #1

4
1
0
0
0

说明/提示

样例 1 解释

样例输入描述了题目描述中举的例子，点的编号从 1 至 5 依次表示草，牛，羊，狼，小强。

数据规模与约定

• 对于 50%的数据，保证 n≤10^4。
• 对于 100%的数据，保证 1≤n≤655341，1≤ai,j≤n，图上的边数不超过 10^6，且图上不存在环。

 

开始想到由一个节点去推它能导致哪些节点消失，然而，这样是无法实现的，因为一个点灭绝需要他所有的食物都消失，所以我们想要把每个点挂在能够直接导致它灭绝的点的下面，也就是它所有的食物的lca，然后去统计这棵树的每个子树的size-1

重点就变成了求DAG上的lca了 倍增就好了

 

代码

#include<bits/stdc++.h>
using namespace std;
const int maxn=65540;
const int maxm=1e6+5;
int n,head[maxn],h[maxn];
struct edge
{
	int nxt,to;
}e[maxm<<1],m[maxm<<1];
int cnt,tot,inin[maxn];
void add(int x ,int y)
{
	e[++cnt].nxt=head[x];
	e[cnt].to=y;
	head[x]=cnt;
}
int f[maxn],fa[maxn][20];
void addd(int x,int y)
{
	m[++tot].nxt=h[x];
	m[tot].to=y;
	h[x]=tot;
}
int dep[maxn];
int lca(int x,int y)
{
	if(dep[x]>dep[y]) swap(x,y);
	for(int i=18;i>=0;i--)
		if(dep[y]-dep[x]>=(1<<i))
			y=fa[y][i];
	if(x==y)  return x;
	for(int i=18;i>=0;i--)
		if(fa[x][i]!=fa[y][i])
			x=fa[x][i],y=fa[y][i];
	return fa[x][0];
}
void topo()
{
	queue <int> q;
	memset(f,-1,sizeof(f));
	for(int i=1;i<=n;i++)
		if(!inin[i])
		{
			q.push(i);
			f[i]=0;
		}
	while(!q.empty())
	{
		int u=q.front();
		q.pop();
		addd(f[u],u);
		dep[u]=dep[f[u]]+1;
		fa[u][0]=f[u];
		for(int i=1;i<=18;i++)
			fa[u][i]=fa[fa[u][i-1]][i-1];
		for(int i=head[u];i;i=e[i].nxt)
		{
			int to=e[i].to;
			if(f[to]==-1) f[to]=u;
			else f[to]=lca(f[to],u);
			inin[to]--;
			if(!inin[to]) q.push(to);		
		}
	}
}
int siz[maxn];
void dfs(int x)
{
	siz[x]=1;
	for(int i=h[x];i;i=m[i].nxt)
	{
		int to=m[i].to;
		dfs(to);
		siz[x]+=siz[to];
	}
}
int main()
{
	freopen("disaster.in","r",stdin);
	freopen("disaster.out","w",stdout);	
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		int x; scanf("%d",&x);
		while(x)
		{
			add(x,i); inin[i]++;
			scanf("%d",&x);
		}
	}
	topo();
	dfs(0);
	for(int i=1;i<=n;i++)
		printf("%d\n",siz[i]-1);
	return 0;
}