PTA天梯赛周训练赛补题 BFS模板题 学习vector用法 复习经典BFS(复习queue)

											7-14 喊山 (30分)

一个山头呼喊的声音可以被临近的山头同时听到。题目假设每个山头最多有两个能听到它的临近山头。给定任意一个发出原始信号的山头，本题请你找出这个信号最远能传达到的地方。

输入格式：
输入第一行给出3个正整数n、m和k，其中n（≤10000）是总的山头数（于是假设每个山头从1到n编号）。接下来的m行，每行给出2个不超过n的正整数，数字间用空格分开，分别代表可以听到彼此的两个山头的编号。这里保证每一对山头只被输入一次，不会有重复的关系输入。最后一行给出k（≤10）个不超过n的正整数，数字间用空格分开，代表需要查询的山头的编号。

输出格式：
依次对于输入中的每个被查询的山头，在一行中输出其发出的呼喊能够连锁传达到的最远的那个山头。注意：被输出的首先必须是被查询的个山头能连锁传到的。若这样的山头不只一个，则输出编号最小的那个。若此山头的呼喊无法传到任何其他山头，则输出0。

输入样例：

7 5 4
1 2
2 3
3 1
4 5
5 6
1 4 5 7

输出样例：

2
6
4
0

AC代码

#include <bits/stdc++.h>

using namespace std;

vector<int>graph[10001];
int sum[10001];
int vis[10001];
int n, Max;

void find(int x)
{
	queue<int>q;
	q.push(x);
	sum[x]++;
	Max = 1;
	vis[x] = 1;
	while (!q.empty())
	{
		int top = q.front();
		for (int i = 0; i < graph[top].size(); i++)
		{
			int num = graph[top][i];
			if (vis[num] == 0)
			{
				vis[num] = 1;
				q.push(num);
				sum[num] = sum[top] + 1;
				if (sum[num] > Max) {
					Max = sum[num];
				}
			}
		}
		q.pop();
	}
}

int main()
{
	int m, k, u, v, x;
	cin >> n >> m >> k;
	for (int i = 0; i < m; i++)
	{
		cin >> u >> v;
		graph[u].push_back(v);
		graph[v].push_back(u);
	}
	while (k--)
	{
		memset(sum, 0, sizeof(sum));
		memset(vis, 0, sizeof(vis));
		cin >> x;
		find(x);
		for (int j = 1; j <= n; j++)
		{
			if (sum[j] == Max) {
				if (Max != 1)
				{
					cout << j << endl;
					break;
				}
				else if (Max == 1)
				{
					printf("0\n");
					break;
				}
			}
		}
	}
}

参考 C++vector的使用总结及常用vector操作

参考 C++容器queue的用法整理

参考 C++ stack用法