hihoCoder1175 拓扑排序·二 拓扑排序

题目链接：

http://hihocoder.com/problemset/problem/1175

题意：

n个点，m条边，有k个点是病毒源，这些病毒源可以传递给所有相邻节点一次，问最终病毒的总数

思路：

考虑每个点最终的病毒数是多少，就是它的所有前驱结点的病毒数的和，

按照拓扑排序的顺序去扩展病毒数量，这样相当于所有节点只走了一遍就出了结果

代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define MS(a) memset(a,0,sizeof(a))
#define MP make_pair
#define PB push_back
const int INF = 0x3f3f3f3f;
const ll INFLL = 0x3f3f3f3f3f3f3f3fLL;
inline ll read(){
    ll x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
//////////////////////////////////////////////////////////////////////////
const int maxn = 1e5+10;
const int mod = 142857;

int a[maxn],in[maxn];
vector<int> g[maxn];

int main(){
    int n,m,k;
    cin >> n >> m >> k;
    for(int i=1; i<=k; i++){
        int u = read();
        a[u]++;
    }
    for(int i=0; i<m; i++){
        int u,v; scanf("%d%d",&u,&v);
        in[v]++;
        g[u].push_back(v);
    }

    queue<int> q;
    for(int i=1; i<=n; i++){
        if(in[i] == 0) q.push(i);
    }
    int ans = 0;
    while(!q.empty()){
        int u = q.front(); q.pop();
        ans = (ans+a[u])%mod;
        for(int i=0; i<(int)g[u].size(); i++){
            int v = g[u][i];
            in[v]--;
            a[v] = (a[v]+a[u])%mod;
            if(in[v] == 0) q.push(v);
        }
    }
    cout << ans << endl;


    return 0;
}