洛谷 P2597 [ZJOI2012]灾难（支配树/必经点）

链接：https://www.luogu.org/problem/P2597

题意：给你一个n，代表n种生物。接下来n行，每行给出生物i可以吃的物种，每行以0结束。对于每种生物i，如果它灭绝，求多少物种也会因为没有食物而灭绝。

思路：题目声明是DAG（有向无环图），那么这个题很明显是个简单支配树（关于必经点）的问题。建立一个超级源点0（可以把它当作太阳），如果它没了，那么所有物种都会灭绝，也就是说他是所有物种的必经点，对应于题目，0的灾难值就是n。题目说是DAG，但没保证图一定联通，也就是说这个图可能是一个森林。这也是引入0点的原因，我们把出度为0的点与0相连（对应于题目中描述的生产者），这样就保证了图为一颗树。显然生产者是大多数生物的必经点（即它如果灭绝，许多生物会跟着灭绝）。我们建图时，将边的方向设为由吃者指向被吃者。由支配树的性质，支配树的中某点的所有祖先都是其必经点。由于我们的建图方式，拓扑排序出来的点，肯定越往后排出的点，在支配树中离根节点越近。再由支配树的性质，对于本题，假设一点为u，他指向的点集V肯定都已在支配树中，把点集V中的所有点在支配树中的LCA求出来（假设为w），那么w就是u在支配树中的父亲。依靠这一性质，结合我们建图的方式，我们按拓扑排序的顺序倒着将点加入支配树中，最后每个点在支配树中子树的大小-1（不包括自己）就是答案。（对于生物i，其实答案就是以生物i为必经点的生物的个数。）

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 7e4+10;
int deep[N],f[N][20],in[N],id[N],q[N],he,ta;
int n,root,x,num,u,v,ans[N];
vector<int> g[N],trg[N];
void TopoSort()
{
	num=0;
	he=ta=0;
 	for(int i=1;i<=n;i++)
 		if(!in[i]) q[ta++]=i;
	while(he!=ta)
	{
		u=q[he++];
		id[++num]=u;
		for(int i=0;i<g[u].size();i++)
		{
			v=g[u][i];
			in[v]--;
			if(!in[v])
			{
				q[ta++]=v;
			}
		}
	}
	return;
}
int Lca(int x,int y)
{
	if(deep[x]<deep[y])
		swap(x,y);
	int diff=deep[x]-deep[y];
	for(int i=17;i>=0;i--)
		if(diff&(1<<i)) x=f[x][i];
	if(x==y) return x;
	for(int i=17;i>=0;i--)
		if(f[x][i]!=f[y][i])
			x=f[x][i],y=f[y][i];
	return f[x][0];
}
void dfs(int u)
{
	//cout<<u<<endl;
	ans[u]=1;
	for(int i=0;i<trg[u].size();i++)
	{
		dfs(trg[u][i]);
		ans[u]+=ans[trg[u][i]];
	}
	return ;
}
int main(void)
{
	//cout<<(1<<16)<<endl;
	//freopen("catas.in","r",stdin);
	//freopen("catas.out","w",stdout);
	//题目描述文件读入读出，其实不用，娘的唬人。 
	root=0;
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		num=0;
		while(~scanf("%d",&x)&&x)
		{
			g[i].push_back(x);
			in[x]++;
			num++;
		}
		if(!num) 	
		{
			//i是生产者，将其与太阳(0)连边 
			//加入支配树中 
			trg[root].push_back(i);
		}	
	}
	TopoSort();
	//倒拓扑排序建图 
	for(int i=n;i>=1;i--)
	{
		
		v=id[i];
		//v是生产者，已经加入支配树中，
		//不用再加入 
		if(!g[v].size()) continue;
		u=g[v][0]; 
		//据性质求 
		for(int j=1;j<g[v].size();j++)
			u=Lca(u,g[v][j]);
		//cout<<v<<" "<<u<<endl;	
		trg[u].push_back(v);
		deep[v]=deep[u]+1;
		f[v][0]=u;
		for(int i=1;i<=17;i++)
			f[v][i]=f[f[v][i-1]][i-1];
	}
	dfs(root);
	for(int i=1;i<=n;i++)
		printf("%d\n",ans[i]-1);
	return 0;
}