可达性统计
给定一张N个点M条边的有向无环图,分别统计从每个点出发能够到达的点的数量。
输入格式
第一行两个整数N,M,接下来M行每行两个整数x,y,表示从x到y的一条有向边。
输出格式
输出共N行,表示每个点能够到达的点的数量。
数据范围
1≤N,M≤30000
输入样例:
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
分析:题目所要求的每个点可以到达的点的数目。其实也就是统计当前点所指向的点的数目+ 所指向的点的可以到达的点的数目 + 1。这一步很明显是一个递推的过程,所以我们需要找到最低层的点,然后一步一步往前推。
因为这是一个有向无环图,所以我们考虑使用拓扑排序,来确定一个合理的从后往前遍历顺序,然后从最低层开始统计。
有关于拓扑排序的相关定义,可以点击此处查看例题和解析。
这里统计数量,我们利用stl的bitset。具体使用方法读者可自行查找。
#include<bits/stdc++.h>
using namespace std;
int n, m;
int in[30010];
vector<int >edge[30010];
vector<int >topop;
bitset<30010> bit[30010];
void top_order() { //拓扑排序
queue<int >que;
for (int i = 1; i <= n; i++) {
if (in[i] == 0) que.push(i);
}
while(!que.empty()) {
int top = que.front();
topop.push_back(top);
que.pop();
for (int i = 0; i < edge[top].size(); i++) {
in[edge[top][i]]--;
if (in[edge[top][i]] == 0) que.push(edge[top][i]);
}
}
}
void count() {
for (int i = topop.size() - 1; i >= 0; i--) {
int x = topop[i];
bit[x].reset();
bit[x][x] = 1;
for (int j = 0; j < edge[x].size(); j++) {
bit[x] = bit[x] | bit[edge[x][j]];
}
}
for (int i = 1; i <= n; i++) {
cout << bit[i].count() << "\n";
}
}
int main () {
cin >> n >> m;
int a, b;
memset(in, 0, sizeof in);
for (int i = 1; i <= m; i++) {
cin >> a >> b;
edge[a].push_back(b);
in[b]++;
}
top_order();
count();
return 0;
}