HIHO #1175 : 拓扑排序·二

题目链接

#include<bits/stdc++.h>
using namespace std;
#define cl(a,b) memset(a,b,sizeof(a))
#define LL long long
#define pb push_back
#define gcd __gcd

#define For(i,j,k) for(int i=(j);i<k;i++)
#define lowbit(i) (i&(-i))
#define _(x) printf("%d\n",x)

const int maxn = 1e5+10;
const int inf  = 1 << 28;

int n,m,k;

int val[maxn];
int in[maxn];

vector<int> G[maxn];
int mod = 142857;
int t_sort(){
    stack<int> s;
    int ans = 0;
    for(int i=1;i<=n;i++)if(in[i]==0)s.push(i);

    while(!s.empty()){
        int u = s.top();s.pop();
        for(int i=0;i<G[u].size();i++){
            int v = G[u][i];
            val[v]+=val[u]%mod;
            if(--in[v]==0){s.push(v);}
        }
    }
    for(int i=1;i<=n;i++)ans+=val[i],ans%=mod;
    return ans;
}
int main(){
    cin>>n>>m>>k;
    int ai;
    for(int i=1;i<=k;i++)cin>>ai,val[ai]=1;
    for(int i=0;i<m;i++){
        int x,y;
        cin>>x>>y;
        G[x].pb(y);
        in[y]++;
    }
    cout<<t_sort()<<endl;

    return 0;
}