洛谷 P1038 神经网络（图的拓扑排序）

图的拓扑排序
本题要点：
1、n <= 100, 很小，用邻接矩阵来存图。 in_deg[MaxN], out_deg[MaxN] 分别表示点的入度，出度。
2、第一步先找到入度为0 的点，全部加入队列中。注意，这部分的点的 转态值c 不需要计算。
因此，用 vis 数组来区分。
3、 然后按照求一般拓扑排序的方式来求（入度为0 的点加入队列），
这里多了一个，计算每个点的 转态值 c。
4、用 set<pair<int, int> > ans 来存答案，自动排序，处理简单。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <queue>
#include <vector>
#include <set>
using namespace std;
const int MaxN = 110;
int mp[MaxN][MaxN], w[MaxN][MaxN];
int c[MaxN], u[MaxN], in_deg[MaxN], out_deg[MaxN];
bool vis[MaxN];
int n, m;
set<pair<int, int> > ans;

void solve()
{
	//先把入度全是 0 的点全部入队
	queue<int> q;
	for(int i = 1; i <= n; ++i)
	{
		if(in_deg[i] == 0)
		{
			q.push(i);
			vis[i] = 1;
		}
	}
	while(q.size())
	{
		int x = q.front(); q.pop();				
		for(int y = 1; y <= n; ++y)	//x 指向的点，入度减1
		{
			if(mp[x][y])
			{
				in_deg[y]--;
				if(in_deg[y] == 0)
				{
					q.push(y);
				}
			}
		}
		if(vis[x] == false)	//输入层的点转态值不需要再计算
		{
			// 计算 x 点的 c 值
			int val = -u[x];
			for(int y = 1; y <= n; ++y)
			{
				if(mp[y][x] && c[y] > 0)
					val += c[y] * w[y][x];
			}
			c[x] = val;
		}
		if(out_deg[x] == 0 && c[x] > 0)	// 点x是输出层
		{
			ans.insert(make_pair(x, c[x]));			
		}
	}
	if(ans.empty())
	{
		printf("NULL\n");
		return;
	}
	for(set<pair<int, int> >::iterator it = ans.begin(); it != ans.end(); ++it)
	{
		printf("%d %d\n", it->first, it->second);	
	}
}

int main()
{
	scanf("%d%d", &n, &m);
	for(int i = 1; i <= n; ++i)
	{
		scanf("%d%d", &c[i], &u[i]);
	}
	int x, y, z;
	for(int i = 0; i < m; ++i)
	{
		scanf("%d%d%d", &x, &y, &z);
		mp[x][y] = 1, w[x][y] = z;
		in_deg[y]++, out_deg[x]++;
	}
	solve();
	return 0;
}

/*
5 6
1 0
1 0
0 1
0 1
0 1
1 3 1
1 4 1
1 5 1
2 3 1
2 4 1
2 5 1
*/

/*
3 1
4 1
5 1
*/