hihocoder1175(拓扑排序)

借鉴了一位大佬的博客: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;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值