hihoCoder #1175 拓扑排序·二

版权声明：希望能帮到弱校的ACMer成长，因为自己是弱校菜鸡~~~~ https://blog.csdn.net/Mr__Charles/article/details/82386323

                            hihoCoder #1175 拓扑排序·二

---

题解：

    穿了个背心的裸题，本题所求的是最后virus的总数。但是，拓扑排序的原过程代码不变，只需添加一个virus的数组来记录每个点的病毒数，最后将1-n累加即为结果。

    拓扑排序的原过程代码不知道的看鄙人的这个博客。

    记得将答案取模......

 

代码：

#include<cstdio>
#include<iostream>
#include<vector>
#include<queue>
#include<algorithm>
using namespace std;
const int MAXN = 1e5+5;
const int MOD = 142857;
int n,m,k,sum;
int indeg[MAXN],virus[MAXN]; 
vector<int> edge[MAXN];
queue<int> Q;

void init(){
	sum = 0;
	for(int i = 1; i <= n; ++i){
		edge[i].clear();
		indeg[i] = 0;
		virus[i] = 0;
	}
	while(!Q.empty()) Q.pop();
}

void topsort(){
	for(int i = 1; i <= n ;++i)
	    if(!indeg[i])
	        Q.push(i);
	while(!Q.empty()){
		int now = Q.front();
		Q.pop();
		for(int i = 0; i < edge[now].size(); ++i){
		    if(--indeg[edge[now][i]] == 0)
		        Q.push(edge[now][i]);       
		    virus[edge[now][i]] = (virus[edge[now][i]] + virus[now]) % MOD;
		}
	}
}

int main(){
	int u,v,vi;
	while(~scanf("%d%d%d",&n,&m,&k)){
		init();
		while(k--){
			scanf("%d",&vi);
			virus[vi]++;
		}
		for(int i = 0; i < m; ++i){
			scanf("%d%d",&u,&v);
			edge[u].push_back(v);
			indeg[v]++;
		}
		topsort();
		for(int i = 1; i <= n; ++i)
		    sum = (sum + virus[i]) % MOD;
		printf("%d\n",sum);
	}
	return 0;
}