Escape The Maze (easy version) 多源最短路，bfs（1700）

题意 ：

• 在一颗树形迷宫中，一共有n个结点，有k个vlad的朋友，分别在$x_1,x_2,...$结点。vlad从1号结点出发，问，在所有朋友都同时移动的同时，vlad是否一定可以走到一个叶子结点（过程中不被朋友抓住）？

思路 ：

• 求的是是否一定可以，也就是不可能被抓住，走到某一个叶子结点的最短路径上的每一个点都比任何朋友走到这个点快，这样可以保证不被抓住
• 多源最短路bfs问题，从所有朋友出发，求出到达i结点的最短路程，再从vlad出发，求是否存在一条到达叶子结点的路径，使得路径上每个点的最短路径都比朋友的短
• 关于bfs ：如果每一次扩展恰好对应一步，那么当一个状态第一次被访问（入队）时，就得到了从起始状态到达该状态的最少步数，就直接入队了，且每个点只入队一次。
• 第二次bfs时，如果当前这个点已经比朋友走到这的最短距离大于等于了，说明不走这个点，直接continue
• 如何表示叶结点 ？ne[h[i]] == -1，注意同时还要判断它不是起点1

#include <iostream>
#include <algorithm>
#include <cstring>
#include <vector>
#include <unordered_set>
#include <math.h>
using namespace std;
 
typedef long long ll;
typedef pair<int, int> PII;
 
#define endl '\n'
#define fi first
#define se second
#define push_back
#define rep(i, l, r) for (ll i = l; i <= r; i ++ )
 
const int N = 2e5 + 10, M = N * 2, inf = 0x3f3f3f3f;
 
int n, k;
int h[N], ne[M], e[M], idx;
int q[N];		// 从朋友出发
int dist[N];	// 所有朋友到i房间的最短时间
bool st[N];
PII mq[N];		// 从vlad出发
 
void add(int a, int b)
{
    e[idx] = b, ne[idx] = h[a], h[a] = idx ++ ;
}
 
void solve()
{
    cin >> n >> k;
    
    // init!
    idx = 0;
    for (int i = 1; i <= n; i ++ ) h[i] = -1, dist[i] = inf, st[i] = false;
    
    int hh = 0, tt = -1;
    for (int i = 1, x; i <= k && cin >> x; i ++ )
    {
        q[ ++ tt] = x;
        st[x] = true;
        dist[x] = 0;
    }
 
    for (int i = 1, u, v; i <= n - 1 && cin >> u >> v; i ++ ) add(u, v), add(v, u);
 
    while (hh <= tt)
    {
        int t = q[hh ++ ];
        for (int i = h[t]; ~i; i = ne[i])
        {
            int j = e[i];
            if (!st[j])  dist[j] = dist[t] + 1, q[ ++ tt] = j, st[j] = true;
        }
    }
    
    // init!!
    for (int i = 1; i <= n; i ++ ) st[i] = false;
    hh = 0, tt = 0;
    mq[0] = {1, 0};
    st[1] = true;
    
    while (hh <= tt)
    {
        auto t = mq[hh ++ ];
        int r = t.fi, d = t.se;
        
        if (d >= dist[r]) continue;
        if (ne[h[r]] == -1 && r != 1 && d < dist[r]) return cout << "Yes\n", void();
        
        for (int i = h[r]; ~i; i = ne[i])
        {
            int j = e[i];
            if (!st[j]) mq[ ++ tt] = {j, d + 1}, st[j] = true;    // 每个点第一次被访问就入队
        }
    }
    
    cout << "No\n";
}
 
int main()
{
    cin.tie(nullptr) -> sync_with_stdio(false);
    
    int _;
    cin >> _;
    while (_ -- )
        solve();
 
    return 0;
}