CCPC-Wannafly Winter Camp Day7 (Div2, onsite) A 迷宫 思维

题解

题意可以理解为一个点上同时只能存在一个人 只要前面没人挡着每秒所有人都可以向出口移动一个单位
考虑两个队的人排队的人合并为一队方式 同一排的人会合并在一起 也就是BFS同层的节点会合并在一起
但是每秒也只能移动一个单位 所以即使前面有很大的空位也只能向前走一步
不需要真的记录整个队伍 只需要记录队伍的最后一个人的距离就是答案 使用k记录最后一个人即k = max(d, k + 1)如果同层没人则直接在层的位置 否则会让整个队伍向后延伸

AC代码

#include <stdio.h>
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

const int INF = 0x3f3f3f3f;
const int MAXN = 1e5 + 10;
int a[MAXN], b[MAXN], k;
vector<int> e[MAXN];
int vis[MAXN];

struct node
{
	int x, d;
};
void BFS()
{
	queue<node> q;
	q.push({ 1, 1 });
	while (!q.empty())
	{
		int x = q.front().x, d = q.front().d;
		q.pop();
		vis[x] = 1;
		for (int y : e[x]) if (!vis[y])
		{
			q.push({ y, d + 1 });
			if (a[y]) //有人
				k = max(d, k + 1); //当前层和紧接着上一个位置取max
		}
	}
}
int main()
{
#ifdef LOCAL
	//freopen("C:/input.txt", "r", stdin);
#endif
	int n;
	cin >> n;
	for (int i = 1; i <= n; i++)
		scanf("%d", &a[i]);
	for (int i = 1; i < n; i++)
	{
		int u, v;
		scanf("%d%d", &u, &v);
		e[u].push_back(v), e[v].push_back(u);
	}
	BFS();
	cout << k << endl;

	return 0;
}