hihocoder 1175 拓扑排序·二

题目链接：拓扑排序·二

题目大意：有一张无环的有向图，刚开始有一些结点有病毒，他会把他所有的病毒传染到后继结点，问最后所有点的病毒总数

题目思路：拓扑排序，从入度为零的结点开始，删的时候传递结点病毒数就好了

#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;

vector<int>vec[maxn];
int InDeg[maxn],virus[maxn];
queue<int>q;
int T,n,m,k,u,v,x,ans;

void topsort(){
    while(!q.empty()) q.pop();
    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 < vec[now].size();i++){
            if(--InDeg[vec[now][i]] == 0) q.push(vec[now][i]);
            virus[vec[now][i]] = (virus[vec[now][i]]+virus[now])%mod;
        }
    }
}

int main(){
    while(~scanf("%d%d%d",&n,&m,&k)){
        for(int i = 1;i <= n;i++) vec[i].clear();
        memset(InDeg,0,sizeof(InDeg));
        memset(virus,0,sizeof(virus));
        while(k--){
            scanf("%d",&x);
            virus[x]++;
        }
        while(m--){
            scanf("%d%d",&u,&v);
            vec[u].push_back(v);
            InDeg[v]++;
        }
        topsort();
        for(int i = 1;i <= n;i++) ans = (ans+virus[i])%mod;
        printf("%d\n",ans);
    }
    return 0;
}