树与图的遍历

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;
}