HihoCoder - 1175 拓扑排序·二 拓扑排序、BFS

拓扑排序·二

时间限制: 10000ms

单点时限: 1000ms

内存限制: 256MB

描述

小Hi和小Ho所在学校的校园网被黑客入侵并投放了病毒。这事在校内BBS上立刻引起了大家的讨论，当然小Hi和小Ho也参与到了其中。从大家各自了解的情况中，小Hi和小Ho整理得到了以下的信息：

• 校园网主干是由N个节点(编号1..N)组成，这些节点之间有一些单向的网路连接。若存在一条网路连接(u,v)链接了节点u和节点v，则节点u可以向节点v发送信息，但是节点v不能通过该链接向节点u发送信息。
• 在刚感染病毒时，校园网立刻切断了一些网络链接，恰好使得剩下网络连接不存在环，避免了节点被反复感染。也就是说从节点i扩散出的病毒，一定不会再回到节点i。
• 当1个病毒感染了节点后，它并不会检查这个节点是否被感染，而是直接将自身的拷贝向所有邻居节点发送，它自身则会留在当前节点。所以一个节点有可能存在多个病毒。
• 现在已经知道黑客在一开始在K个节点上分别投放了一个病毒。

举个例子，假设切断部分网络连接后学校网络如下图所示，由4个节点和4条链接构成。最开始只有节点1上有病毒。

最开始节点1向节点2和节点3传送了病毒，自身留有1个病毒：

其中一个病毒到达节点2后，向节点3传送了一个病毒。另一个到达节点3的病毒向节点4发送自己的拷贝：

当从节点2传送到节点3的病毒到达之后，该病毒又发送了一份自己的拷贝向节点4。此时节点3上留有2个病毒：

最后每个节点上的病毒为：

小Hi和小Ho根据目前的情况发现一段时间之后，所有的节点病毒数量一定不会再发生变化。那么对于整个网络来说，最后会有多少个病毒呢？

提示：拓扑排序的应用

输入

第1行：3个整数N,M,K，1≤K≤N≤100,000，1≤M≤500,000

第2行：K个整数A[i]，A[i]表示黑客在节点A[i]上放了1个病毒。1≤A[i]≤N

第3..M+2行：每行2个整数 u,v，表示存在一条从节点u到节点v的网络链接。数据保证为无环图。1≤u,v≤N

输出

第1行：1个整数，表示最后整个网络的病毒数量 MOD 142857

样例输入

4 4 1
1
1 2
1 3
2 3
3 4

样例输出

6

Source

HihoCoder - 1175

My Solution

题意：初始有k个节点上有一个病毒，病毒会沿着有向边向向来的节点复制传染一次，图保持无环，最后图上病毒的分布必定会到达稳定状态，求达到稳定状态时，图上的病毒数量。

拓扑排序、BFS

因为保证无环，所以可以BFS， 每次选择入度为0的节点，把病毒数量传递下去。

复杂度 O(n+m)

#include <iostream>
#include <cstdio>
#include <vector>
#include <queue>
#include <cstring>
using namespace std;
typedef long long LL;
typedef pair<int, int> ii;
const int MAXN = 1e5 + 8;
const LL MOD = 142857;

vector<int> sons[MAXN];
int deg[MAXN];
LL cnt[MAXN];
queue<ii> que;

inline LL mod(LL x)
{
    return x - x / MOD * MOD;
}
LL sum = 0;

inline void bfs()
{
    int fa, u, v, sz, i;
    while(!que.empty()){
        u = que.front().first;
        fa = que.front().second;
        que.pop();
        sz = sons[u].size();
        for(i = 0; i < sz; i++){
            v = sons[u][i];
            if(v == fa) continue;
            deg[v]--;
            cnt[v] = mod(cnt[v] + cnt[u]);
            if(deg[v] == 0){
                que.push(ii(v, u));
                sum = mod(sum + cnt[v]);
            }
        }
    }
}

int main()
{
    #ifdef LOCAL
    freopen("2.in", "r", stdin);
    //freopen("2.out", "w", stdout);

    #endif // LOCAL
    ios::sync_with_stdio(false); cin.tie(0);

    int n, m, k, ai, i, j, u, v, sz;
    memset(deg, 0, sizeof deg);
    memset(cnt, 0, sizeof cnt);
    cin >> n >> m >> k;
    for(i = 1; i <= k; i++){
        cin >> ai;
        cnt[ai] = 1;
    }
    for(i = 0; i < m; i++){
        cin >> u >> v;
        sons[u].push_back(v);
    }
    for(i = 1; i <= n; i++){
        sz = sons[i].size();
        for(j = 0; j < sz; j++){
            deg[sons[i][j]]++;
        }
    }
    for(i = 1; i <= n; i++){
        if(deg[i] == 0){
            que.push(ii(i, -1));
            sum += cnt[i];
        }
    }
    bfs();
    while(sum < 0) sum += MOD;
    cout << sum << endl;
    return 0;
}

  Thank you!

                                                                                                                                             ------from ProLights