【拓扑排序+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;
}
上一篇: CH 2101 可达性统计 拓扑排序+bitset
下一篇: STL中的priority_queue