【深基18.例3】查找文献

链接

题目描述

小 K 喜欢翻看洛谷博客获取知识。每篇文章可能会有若干个（也有可能没有）参考文献的链接指向别的博客文章。小 K 求知欲旺盛，如果他看了某篇文章，那么他一定会去看这篇文章的参考文献（如果他之前已经看过这篇参考文献的话就不用再看它了）。

假设洛谷博客里面一共有 $n(n\le 10^5)$ 篇文章（编号为 1 到 $n$）以及 $m(m\le 10^6)$ 条参考文献引用关系。目前小 K 已经打开了编号为 1 的一篇文章，请帮助小 K 设计一种方法，使小 K 可以不重复、不遗漏的看完所有他能看到的文章。

这边是已经整理好的参考文献关系图，其中，文献 X → Y 表示文章 X 有参考文献 Y。不保证编号为 1 的文章没有被其他文章引用。

请对这个图分别进行 DFS 和 BFS，并输出遍历结果。如果有很多篇文章可以参阅，请先看编号较小的那篇(因此你可能需要先排序)。

输入格式

共 $m+1$ 行，第 1 行为 2 个数，$n$ 和 $m$，分别表示一共有 $n(n\le 10^5)$ 篇文章（编号为 1 到 $n$）以及$m(m\le 10^6)$ 条参考文献引用关系。

接下来 $m$ 行，每行有两个整数 $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

思路：

根据题目可知，我们只需要构造邻接表，然后进行排序，在进行DFS 以及BFS 即可

代码如下：

#include<map>
#include<set>
#include<stack>
#include<cmath>
#include<queue>
#include<string>
#include<bitset>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<numeric>
#define endl '\n'
using namespace std;

typedef long long ll;
typedef pair<int, int>PII;
const int N=3e5+10;
const int MOD=998244353;
const int INF=0X3F3F3F3F;
const int dx[]={-1,1,0,0,-1,-1,+1,+1};
const int dy[]={0,0,-1,1,-1,+1,-1,+1};
const int M = 1e4 + 10;

vector<int> v[N];
bitset<N> st;

void dfs(int u)
{
	if(!v[u].size())//判断结束的情况
	{
		return ;
	}
	else {
		sort(v[u].begin(), v[u].end());// 因为要符合题目的要求
		for(int i = 0; i < v[u].size(); i ++)
		{
			if(!st[v[u][i]])
			{
				st[v[u][i]] = 1;
				cout << v[u][i] << " ";
				dfs(v[u][i]);//因为只能输出一次所以不用在回溯了
            }
        }
	}
}
int n, m;
int main()
{
	cin >> n >> m;
	for(int i = 1; i <= m; i ++)
	{
		int x, y;
		cin >> x >> y;
		v[x].push_back(y);
	}
	cout << 1 << " ";
	st[1] = 1;//已经用过了
	dfs(1);
	cout << endl;
	st.reset();
	st[1] = 1;
	queue<int>q;
	cout << 1 << " ";
	q.push(1);
	while(!q.empty()){
		auto x = q.front();
		q.pop();
//		if(!st[x]) continue;
//		st[x] = 1;
		sort(v[x].begin(), v[x].end());
		for(int i = 0; i < v[x].size(); i ++)
		{
			if(!st[v[x][i]])
			{
				cout << v[x][i] << " ";
				st[v[x][i]] = 1;
				q.push(v[x][i]);
			}
		}
	}
	return 0;
}