欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

【拓扑排序+bitset】可达性统计

程序员文章站 2022-07-12 16:39:31
...
/*
可达性统计
时间限制: 1 Sec  内存限制: 128 MB
题目描述
给定一张N个点M条边的有向无环图,分别统计从每个点出发能够到达的点的数量。N,M≤30000。

输入
第一行两个整数N,M,接下来M行每行两个整数x,y,表示从x到y的一条有向边。

输出
共N行,表示每个点能够到达的点的数量。

样例输入
复制样例数据
10 10
3 8
2 3
2 5
5 9
5 9
2 3
3 9
4 8
2 10
4 9
样例输出
1
6
3
3
2
1
1
1
1
1
*/
#include <iostream>
#include <vector>
#include <list>
#include <queue>
#include <bitset>

#define N 30002
using namespace std;
int n, m;
vector<int> G[N];
vector<int> topo;
int indeg[N];
bitset<N> bit[N];

void tsort() {
    queue<int> q;
    for (int u = 1; u <= n; ++u) {
        if (indeg[u] == 0) {
            q.push(u);
        }
    }
    while (!q.empty()) {
        int u = q.front();
        q.pop();
        topo.push_back(u);
        for (int i = 0; i < G[u].size(); ++i) {
            int v = G[u][i];
            indeg[v]--;
            if (indeg[v] == 0) {
                q.push(v);
            }
        }
    }
}

void solve() {
    for (int i = topo.size() - 1; i >= 0; i--) {
        int x = topo[i];
        bit[x].reset();
        bit[x][x] = true;
        for (int j = 0; j < G[x].size(); j++) {
            bit[x] = bit[x] | bit[G[x][j]];
        }
    }
    for (int i = 1; i <= n; i++) {
        cout << bit[i].count() << endl;
    }
}

int main() {
    cin >> n >> m;
    int u, v;
    for (int i = 0; i < m; ++i) {
        cin >> u >> v;
        G[u].push_back(v);
        indeg[v]++;
    }
    tsort();
    solve();
    return 0;
}