借鉴了一位大佬的博客:http://blog.csdn.net/nemaleswang/article/details/78006376
题目链接:http://hihocoder.com/problemset/problem/1175
题目大意:有一张无环的有向图,刚开始有一些结点有病毒,他会把他所有的病毒传染到后继结点,问最后所有点的病毒总数。
思路:可以把入度为0的点,入队列,然后一条条地删除边并更新每个点的病毒数目,继续把入度为0的点入队列。这样这就只遍历了一遍,我开始就用普通的dfs做的,遍历了无数次,就超时了。
代码:
#include <map>
#include <set>
#include <cmath>
#include <queue>
#include <stack>
#include <vector>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 1e5+10;
const int mod = 142857;
int main(){
int n,m,k,a[maxn],inDeg[maxn],u,v,viru[maxn];
vector<int>vec[maxn];
queue<int>q;
while(~scanf("%d%d%d",&n,&m,&k)){
for(int i = 0;i <= n;i++){
vec[i].clear();
viru[i] = 0;
}
while(!q.empty()) q.pop();
while(k--){
int x;
scanf("%d",&x);
viru[x]++;
}
while(m--){
scanf("%d%d",&u,&v);
inDeg[v]++;
vec[u].push_back(v);
}
for(int i = 1;i <= n;i++){
if(inDeg[i] == 0) q.push(i);
}
while(!q.empty()){
int now = q.front();
q.pop();
for(int i = 0;i < vec[now].size();i++){
if(--inDeg[vec[now][i]] == 0) q.push(vec[now][i]);
viru[vec[now][i]] = (viru[vec[now][i]]+viru[now])%mod;
}
}
int ans = 0;
for(int i = 1;i <= n;i++) ans = (ans+viru[i])%mod;
printf("%d\n",ans);
}
return 0;
}