164. 可达性统计
一开始直接暴力搜,结果TLE了
#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>
using namespace std;
const int N = 3e4 + 5;
bool vis[N];
vector<int> G[N];
int cnt;
void dfs(int u)
{
if (!vis[u]){
cnt++;
vis[u] = 1;
for (int i = 0; i < G[u].size(); i++)
if (G[u][i])
dfs(G[u][i]);
}
}
int main(void)
{
int n, m;
int u, v;
scanf("%d%d", &n, &m);
for (int i = 0; i < m; i++){
scanf("%d%d", &u, &v);
G[u].push_back(v);//单向边
}
for (int i = 1; i <= n; i++){
cnt = 0;
memset(vis, 0, sizeof vis);
dfs(i);
printf("%d\n", cnt);
}
return 0;
}
因为是有向无环图,可以使用拓扑排序
题解的做法,先进行拓扑排序,再逆序进行递推
用 f 数组来记录每个点能够访问的点,1来表示能够访问,0表示不能访问
拓扑排序之后,因为拓扑序列只能由前面的点来访问后面的点,不能正序访问,因为更新时后面的点还没有更新。
逆序访问时,用到的点都已经被更新过
#include <cstdio>
#include <cstring>
#include <queue>
#include <bitset>
#include <algorithm>
using namespace std;
const int N = 3e4 + 5;
int t, n, m;
int in[N]; //每个节点的度
queue<int> q;
vector<int> edge[N], topo;//topo为拓扑序列
bitset<N> f[N];
void init()
{
for (int i = 1; i <= n; i++){
edge[i].clear();
in[i] = 0;
}
while (!q.empty())
q.pop();
topo.clear();
}
bool toposort()
{
for (int i = 1; i <= n; i++) //n为节点的总数
if (in[i] == 0)
q.push(i); //将度为0的节点放入队列
while (!q.empty()) {
int p = q.front(); q.pop(); //选一个度为0的点出队
topo.push_back(p);
for (int i = 0; i < edge[p].size(); i++) {
int y = edge[p][i];
in[y]--;
if (in[y] == 0)
q.push(y);
}
}
if (topo.size() == n)
return true;
else
return false;
}
int main(void)
{
int a, b;
scanf("%d%d", &n, &m);
init();
while (m--){
scanf("%d%d", &a, &b);
edge[a].push_back(b);
in[b]++;
}
toposort();
for (int i = n - 1; i >= 0; i--){
int j = topo[i];
f[j][j] = 1;
for (int k = 0; k < edge[j].size(); k++)
f[j] |= f[edge[j][k]];
}
for (int i = 1; i <= n; i++)
printf("%d\n", f[i].count());
return 0;
}