题目链接:hihocoder
注意:
病毒对所有的出度边都复制一份自身的内容
当自身入度为0之后,病毒的数量就不会再改变。
坑点:
病毒数量求和的时候,要边加边取模。
#include<cstdio>
#include<vector>
#include<queue>
#include<algorithm>
#include<cstring>
using namespace std;
const int maxn = 1e5+10;
const int mod = 142857;
vector<int> G[maxn];
int inDeg[maxn];
int virus[maxn];
int n,m,k;
void topsort(){
queue<int> q;
for(int i=1;i<=n;i++){
if(inDeg[i] == 0){
q.push(i);
}
}
while(!q.empty()){
int tmp = q.front();
q.pop();
for(int i=0;i<G[tmp].size();i++){
int v = G[tmp][i];
inDeg[v]--;
if(inDeg[v]==0){
q.push(v);
}
virus[v] = (virus[v] + virus[tmp]) % mod;
}
G[tmp].clear();
}
}
int main(){
int u,v;
for(int i=0;i<maxn;i++){
G[i].clear();
}
memset(inDeg,0,sizeof(inDeg));
memset(virus,0,sizeof(virus));
scanf("%d%d%d",&n,&m,&k);
while(k--){
scanf("%d",&v);
virus[v]=1;
}
for(int i=0;i<m;i++){
scanf("%d%d",&u,&v);
G[u].push_back(v);
inDeg[v]++;
}
topsort();
int sum=0;
for(int i=1;i <= n;i++){
sum = (sum + virus[i])% mod ;
}
printf("%d\n",sum);
return 0;
}