题解 1707: 数据结构-有向无环图的拓扑排序

#include<iostream>
#include<stack>
using namespace std;
int n,topo[100]={0},indegree[100]={0},a[100][100];
void FindInDegree()
{
	for(int i=0;i<n;i++)
		for(int j=0;j<n;j++)
			if(a[i][j])
				indegree[j]++;
}
bool TopoSort()
{
	FindInDegree();
	stack<int> s;
	int m=0;
	for(int i=0;i<n;i++)
		if(indegree[i]==0)
			s.push(i);
	while(!s.empty())
	{
		int i=s.top();
		topo[m++]=i;
		s.pop();
		for(int j=0;j<n;j++)
			if(a[i][j]==1)
			{
				indegree[j]--;
				if(indegree[j]==0)
					s.push(j);
			}
	}
	if(m<n)
		return false;
	else
		return true;
}
int main()
{
	cin>>n;
	for(int i=0;i<n;i++)
		for(int j=0;j<n;j++)
			cin>>a[i][j];
	if(!TopoSort())
	 	cout<<"ERROR";
	else
		for(int i=0;i<n;i++)
			cout<<topo[i]<<" ";
	cout<<endl;
	return 0;
}