洛谷2002 消息扩散 (tarjan)

b站上搜到一个比较好的视频：https://www.bilibili.com/video/av7330663/

算法的思想是用dfn(时间戳，也可以看作每个点访问的顺序)、low(标记可达的最小的时间戳)、在加上一个栈存储辅助，通过dfs的过程来判断强连通分量。

dfn标记这个点在dfs的过程中第几次访问到，所以这个点一旦确定就不用再更改了， 只需要比较更新l这个点的low值就可以了。

---

题目描述
有n个城市，中间有单向道路连接，消息会沿着道路扩散，现在给出n个城市及其之间的道路，问至少需要在几个城市发布消息才能让这所有n个城市都得到消息。

输入格式
第一行两个整数n,m表示n个城市，m条单向道路。

以下m行，每行两个整数b,e表示有一条从b到e的道路，道路可以重复或存在自环。

输出格式
一行一个整数，表示至少要在几个城市中发布消息。

输入输出样例
输入
5 4
1 2
2 1
2 3
5 1
输出
2

#include <iostream>
#include <cstdio>
#include <cstring>
#include <stack>
#include <algorithm>
using namespace std;

const int MAXN = 100100;
struct Edge{	//邻接表存图 
	int to, nxt;
}e[500100];
int head[MAXN], dfn[MAXN], low[MAXN];
int bel[MAXN], t[MAXN], x[500100], y[500100];
bool vis[MAXN];	//vis标记是否在栈中 
int cnt, n, m, tot, ans, num;
stack<int> s;

void add(int u, int v){
//邻接表加边 
	cnt++;	//cnt-->边的编号 
	e[cnt].to = v;
	e[cnt].nxt = head[u];
	head[u] = cnt; 
}
void tarjan(int u){
//tarjan：基于dfs 
	low[u] = dfn[u] = ++tot;
	s.push(u);
	vis[u] = true;
	for (int i=head[u]; i; i=e[i].nxt){
		int v = e[i].to;
		if (dfn[v]==0){
			tarjan(v);	//dfs过程 
			low[u] = min(low[u], low[v]);
		}else if (vis[v]){
			low[u] = min(low[u], dfn[v]);
		}
	}
	if (dfn[u]==low[u]){
		int now=-1;
		num++;	//强连通分量个数编号 
		while (now!=u){
			now = s.top();
			s.pop();
			bel[now] = num;
			vis[now] = false;
		}
	}
}
int main(){
	scanf("%d%d", &n, &m);
	for (int i=1; i<=m; i++){
		scanf("%d%d", &x[i], &y[i]);
		add(x[i], y[i]);
	}
	for (int i=1; i<=n; i++){
	//可能一次遍历不完所有点 
		if (dfn[i]==0){	
			tarjan(i);
		}
	}
	//tarjan结束后，所有强连通分量被分开。但是存在有两个强连通分量，a->b, 但b!->a 
	for (int i=1; i<=m; i++){
		if (bel[x[i]]!=bel[y[i]]){
			t[bel[y[i]]]++;	//如果x[i]可达y[i],那么y[i]所在强连通分量入度++ 
		}
	} 
	for (int i=1; i<=num; i++){
		if (t[i]==0){	//对于入度为0的强连通分量，需要给他消息 
			ans++;
		}
	}
	printf("%d\n", ans);
	return 0;
}