【C++】洛谷P3916 图的遍历

图的遍历

题目描述

给出 $N$ 个点，$M$ 条边的有向图，对于每个点 $v$，求 $A(v)$ 表示从点 $v$ 出发，能到达的编号最大的点。

输入格式

第 $1$ 行 $2$ 个整数 $N,M$，表示点数和边数。

接下来 $M$ 行，每行 $2$ 个整数 $U_i,V_i$，表示边 $(U_i,V_i)$。点用 $1,2,\dots ,N$ 编号。

输出格式

一行 $N$ 个整数 $A(1),A(2),\dots ,A(N)$。

样例 #1

样例输入 #1

4 3
1 2
2 4
4 3

样例输出 #1

4 4 3 4

提示

• 对于 $60\%$ 的数据，$1\le N,M\le 10^3$。
• 对于 $100\%$ 的数据，$1\le N,M\le 10^5$。

思路： 直接强行从每一个点遍历一次整个图会超时，所以需要换一种方法。
由于这是个有向图，那么在按照图的方向遍历时，例如从a->b时，若经过了b那么就一定可以经过a。所以我们可以将有向图的方向倒置（a->b的图就变成了b->a的图），然后由题意，我们需要寻找编号最大的点，因此可以从最大的点开始遍历，此时遍历的路径上的所有点能到达的最大点就是此时的起始点。
按照这个思路将图遍历一次即可。因为每个点只会被遍历一次，所以不会超时。

代码：

#include <bits/stdc++.h>
using namespace std;
int n, m, ans[100005];
bool vis[100005];
vector<int> vec[100005];
void dfs(int st, int maxx){
    vis[st] = true;
    maxx = max(st, maxx);
    ans[st] = max(maxx, ans[st]);
    for(int i = 0; i < vec[st].size(); i++){
        int to = vec[st][i];
        if(!vis[to]) dfs(to, maxx);
    }
    return;
}
int main(){
    cin >> n >> m;
    for(int i = 1; i <= m; i++){
        int x, y;
        cin >> x >> y;
        vec[y].push_back(x);
    }
    for(int i = 1; i <= n; i++) ans[i] = i;
    for(int i = n; i >= 1; i--) if(!vec[i].empty()) dfs(i,-1);
    for(int i = 1; i <= n; i++) printf("%d ", ans[i]);
    return 0;
}