P2921 [USACO08DEC]在农场万圣节Trick or Treat on the Farm

题目链接：：P2921 [USACO08DEC]在农场万圣节Trick or Treat on the Farm

哈哈，这道题和信息传递很类似。发现拓扑排序和有向图最小环问题真的联系密切。

和信息传递那一题不太一样的是，我们不仅要求最小环的步长，也要求那些不成环的点的步长。其实这也不难，仔细想想，不能成为完整的一个环的话,它所得到糖果数不就是把它从下一个隔间得来的糖果数量再+1吗；

举例

...

 

#include <cstdio>
#include <queue>
#include <stack>
using namespace std;
int P[100010];
int step[100100];
int indegree[100100];
int del[100010];
int DFS(int t,int steps)		//DFS求环的步数,并且把环内所有点的步数更新为环的周长 
{
	if (del[t]==1)
		return step[t]=steps;
	del[t]=1;
	steps++;
	return step[t]=DFS(P[t],steps);
}
void topsort(int n)		//拓扑排序 
{
	int v;
	queue <int> q;
	stack <int> p;
	for (int i=1;i<=n;i++)
		if (indegree[i]==0)
			q.push(i);
	while (!q.empty())
	{
		v=q.front();
		del[v]=1;
		q.pop();
		p.push(v);	//用栈存下不能形成环的点,得到反拓扑序列 
		indegree[P[v]]--;
		if (indegree[P[v]]==0)
			q.push(P[v]);
	}
	for (int i=1;i<=n;i++)	//对 环们 DFS 
	{
		if (del[i]==0)
		{
			DFS(i,0);
		}
	}
	while (!p.empty())	//对反拓扑序列中每一个点求步数. 
	{
		step[p.top()]=step[P[p.top()]]+1;
		p.pop();
	}
} 

int main()
{
	int n;
	scanf ("%d",&n);
	for (int i=1;i<=n;i++)
	{
		scanf ("%d",&P[i]);
		indegree[P[i]]++;
	}
	topsort(n);
	for (int i=1;i<=n;i++)
	{
		printf ("%d\n",step[i]);
	}
	return 0;
}