题目描述
小 K 喜欢翻看洛谷博客获取知识。每篇文章可能会有若干个(也有可能没有)参考文献的链接指向别的博客文章。小 K 求知欲旺盛,如果他看了某篇文章,那么他一定会去看这篇文章的参考文献(如果他之前已经看过这篇参考文献的话就不用再看它了)。
假设洛谷博客里面一共有 n ( n ≤ 1 0 5 ) n(n\le10^5) n(n≤105) 篇文章(编号为 1 到 n n n)以及 m ( m ≤ 1 0 6 ) m(m\le10^6) m(m≤106) 条参考文献引用关系。目前小 K 已经打开了编号为 1 的一篇文章,请帮助小 K 设计一种方法,使小 K 可以不重复、不遗漏的看完所有他能看到的文章。
这边是已经整理好的参考文献关系图,其中,文献 X → Y 表示文章 X 有参考文献 Y。不保证编号为 1 的文章没有被其他文章引用。
请对这个图分别进行 DFS 和 BFS,并输出遍历结果。如果有很多篇文章可以参阅,请先看编号较小的那篇(因此你可能需要先排序)。
输入格式
共 m + 1 m+1 m+1 行,第 1 行为 2 个数, n n n 和 m m m,分别表示一共有 n ( n ≤ 1 0 5 ) n(n\le10^5) n(n≤105) 篇文章(编号为 1 到 n n n)以及 m ( m ≤ 1 0 6 ) m(m\le10^6) m(m≤106) 条参考文献引用关系。
接下来 m m m 行,每行有两个整数 X , Y X,Y X,Y 表示文章 X 有参考文献 Y。
输出格式
共 2 行。
第一行为 DFS 遍历结果,第二行为 BFS 遍历结果。
样例 #1
样例输入 #1
8 9
1 2
1 3
1 4
2 5
2 6
3 7
4 7
4 8
7 8
样例输出 #1
1 2 5 6 3 7 8 4
1 2 3 4 5 6 7 8
代码
#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>
#define endl '\n'
using namespace std;
typedef pair<int, int> PII;
const int N = 100010, M = 1000010;
int n, m;
int h[N], e[M], ne[M], idx;
bool st[N];
PII edges[M];
void add(int a, int b) {
e[idx] = b;
ne[idx] = h[a];
h[a] = idx++;
}
// 自定义建边操作的排序规则,目的是为了在遍历时先输出编号较小的节点
// 按起点升序,终点降序
// 因为在邻接表中,每新建一条边,都是从表头插入
// 按终点降序可以保证编号较小的节点先被遍历到
bool compare(PII a, PII b) {
if (a.first != b.first) {
return a.first < b.first;
}
else {
return a.second > b.second;
}
}
// DFS
void dfs(int t) {
st[t] = true;
cout << t << " ";
for (int i = h[t]; i != -1; i = ne[i]) {
int j = e[i];
if (!st[j]) {
dfs(j);
}
}
}
// BFS
void bfs() {
queue<int> q;
q.push(1);
st[1] = true;// 不要忘了这个
cout << 1 << " ";
while (q.size()) {
int t = q.front();
q.pop();
for (int i = h[t]; i != -1; i = ne[i]) {
int j = e[i];
if (!st[j]) {
q.push(j);
st[j] = true;
cout << j << " ";
}
}
}
}
int main() {
ios::sync_with_stdio(false), cin.tie(nullptr), cout.tie(nullptr);
memset(h, -1, sizeof h);
cin >> n >> m;
for (int i = 0; i < m; i++) {
cin >> edges[i].first >> edges[i].second;
}
// 给所有建边操作排序
sort(edges, edges + m, compare);
for (int i = 0; i < m; i++) {
int a = edges[i].first, b = edges[i].second;
add(a, b);
}
dfs(1);
cout << endl;
memset(st, false, sizeof st);// dfs之后要重置st数组,将所有点重新标记为没有遍历过
bfs();
cout << endl;
return 0;
}