小Hi和小Ho所在学校的校园网被黑客入侵并投放了病毒。这事在校内BBS上立刻引起了大家的讨论,当然小Hi和小Ho也参与到了其中。从大家各自了解的情况中,小Hi和小Ho整理得到了以下的信息:
举个例子,假设切断部分网络连接后学校网络如下图所示,由4个节点和4条链接构成。最开始只有节点1上有病毒。

最开始节点1向节点2和节点3传送了病毒,自身留有1个病毒:

其中一个病毒到达节点2后,向节点3传送了一个病毒。另一个到达节点3的病毒向节点4发送自己的拷贝:

当从节点2传送到节点3的病毒到达之后,该病毒又发送了一份自己的拷贝向节点4。此时节点3上留有2个病毒:

最后每个节点上的病毒为:

小Hi和小Ho根据目前的情况发现一段时间之后,所有的节点病毒数量一定不会再发生变化。那么对于整个网络来说,最后会有多少个病毒呢?
第1行:3个整数N,M,K,1≤K≤N≤100,000,1≤M≤500,000
第2行:K个整数A[i],A[i]表示黑客在节点A[i]上放了1个病毒。1≤A[i]≤N
第3..M+2行:每行2个整数 u,v,表示存在一条从节点u到节点v的网络链接。数据保证为无环图。1≤u,v≤N
第1行:1个整数,表示最后整个网络的病毒数量 MOD 142857
4 4 1 1 1 2 1 3 2 3 3 4
6
有要取模的地方,注意对所有运算取模,防止有bug。。。
AC代码:
#include <map>
#include <set>
#include <cmath>
#include <deque>
#include <queue>
#include <stack>
#include <cstdio>
#include <cctype>
#include <string>
#include <vector>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
#define LL long long
#define INF 0x7fffffff
using namespace std;
#define MOD 142857
int n, m, k;
const int maxn = 100005;
vector<int> G[maxn];
int indeg[maxn];
int sum[maxn];
int bfs() {
	int ret = 0;
	queue<int> que;
	for(int i = 1; i <= n; i ++) {
		if(indeg[i] == 0) {
			que.push(i);
			ret = (ret + sum[i]) % MOD;
		}
	}
	
	while(!que.empty()) {
		int u = que.front();
		que.pop();
		int d = G[u].size();
		for(int i = 0; i < d; i ++) {
			int v = G[u][i];
			indeg[v] --;
			sum[v] = (sum[v] + sum[u]) % MOD;	//记得要模MOD,不然会溢出。 
			if(indeg[v] == 0) {
				que.push(v);
				ret = (ret + sum[v]) % MOD;
			}
		}
	}
	return ret;
}
int main() {
	while(scanf("%d %d %d", &n, &m, &k) != EOF) {
		for(int i = 1; i <= n; i ++) G[i].clear();
		memset(sum, 0, sizeof(sum));
		memset(indeg, 0, sizeof(indeg));
		for(int i = 0; i < k; i ++) {
			int x;
			scanf("%d", &x);
			sum[x] ++;
		}
		for(int i = 0; i < m; i ++) {
			int u, v;
			scanf("%d %d", &u, &v);
			G[u].push_back(v);
			indeg[v] ++;
		}
		printf("%d\n", bfs());
	}
	return 0;
}
版权声明:本文为博主原创文章,未经博主允许不得转载。
hihoCoder - 1175 - 拓扑排序·二 (拓扑排序的应用)
原文地址:http://blog.csdn.net/u014355480/article/details/47000189