P1038 [NOIP2003 提高组] 神经网络

知识点:拓扑排序,模拟

难度:4

这个题考察的知识点就是拓扑排序+模拟,我采用的方法是建立两个图的写法,建立了一个反图,拓扑排序的时候,每排序一个点,就用反图来计算这个点的c值,思路简单易懂就是代码稍微多了一点,多建了一个图,然后特判只需要判断一下当前点是不是输入层就行了,因为输入层是不需要计算c值的,

花了不到20分钟过了这个题

#include 

using namespace std;

const int N = 1e4 + 5;

int n, m, tot, ver[N], edge[N], nxt[N], head[N];
int tot2, ver2[N], edge2[N], nxt2[N], head2[N];
int indeg[N], c[N], u[N];

void add(int x, int y, int z) {
	ver[++tot] = y; edge[tot] = z;
	nxt[tot] = head[x]; head[x] = tot;
}

void add2(int x, int y, int z) {
	ver2[++tot2] = y; edge2[tot2] = z;
	nxt2[tot2] = head2[x]; head2[x] = tot2;
}

void topo() {
	queue q;
	for (int i = 1; i <= n; i++) {
		if (!indeg[i]) q.push(i);
	}
	while (!q.empty()) {
		int now = q.front(); q.pop();
		if (!c[now]) {
			int sum = 0;
			for (int i = head2[now]; i; i = nxt2[i]) {
				int y = ver2[i];
				if (c[y] > 0) sum += edge2[i] * c[y];
			}
			c[now] = sum - u[now];
		}
		for (int i = head[now]; i; i = nxt[i]) {
			int y = ver[i];
			if (--indeg[y] == 0) q.push(y);
		}
	}
}

int main() {
	cin >> n >> m;
	for (int i = 1; i <= n; i++) {
		cin >> c[i] >> u[i];
	}
	while (m--) {
		int x, y, z;
		cin >> x >> y >> z;
		add(x, y, z);
		add2(y, x, z);
		indeg[y]++;
	}
	topo();
	int ok = 0;
	for (int i = 1; i <= n; i++) {
		if (!head[i] && c[i] > 0) {
			cout << i << ' ' << c[i] << '\n';
			ok = 1;
		}
	}
	if (!ok) cout << "NULL";
	return 0;
}

你可能感兴趣的:(洛谷刷题记录,算法,c++,数据结构)