2017浙工大院赛预赛 F 小妈妈找蝌蚪【dijkstra】

题目描述
青蛙妈妈最近很不放心把蝌蚪宝宝送到幼儿园，但当她买菜回家时，却发现可爱的孩子小蝌蚪走丢了。
小池塘里有很多石头，青蛙家在其中标号为s的石头上。小蝌蚪会移动k分钟，每分钟会出现在任意石头旁边，甚至多次出现在一块石头旁边。但k分钟之后，蝌蚪宝宝就游不动了。
青蛙妈妈第0秒从家所在的石头出发，每分钟移动一次，可以留在原地，也可以跳跃到一块当前可跳跃到的石头上(只能在特定的石头间双向跳跃)。
输入描述:
多组数据。
第一行输入石头个数n，青蛙妈妈可以跳跃的石头对数m，蝌蚪宝宝的活动时间k，青蛙家所在的石头s。
之后输入k个数，其中第i个数代表第i分钟蝌蚪宝宝的位置，编号从i=1开始。
接下来输入m行，每行包括两个数ui，vi，表示青蛙妈妈可以在第ui个和vi个石头间双向跳跃。
输出描述:
请输出青蛙妈妈最少几分钟发现蝌蚪宝宝。
示例1
输入

3 2 2 1
2 1
1 2
1 3
5 5 3 5
3 1 4
1 2
2 3
2 4
3 4
4 5
输出

1
3

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<vector>
#include<queue>
#include<functional>
const int N = 2e5 + 10;
#define INF 0x3f3f3f3f
#define lowbit(o) o&-o
using namespace std;
int a[N];
int i, j, k, nx[N], to[N], q[N], head[N], b[N], x, y, n, m, tot, s, bb;
void add(int x, int y) {
    nx[++tot] = head[x]; head[x] = tot; to[tot] = y;
}
int main()
{
    while (~scanf("%d%d%d%d", &n, &m, &k, &s))
    {
        for (i = 1; i <= n; i++) head[i] = 0; tot = 0;
        for (i = 1; i <= k; i++) scanf("%d", &a[i]);
        for (i = 1; i <= m; i++) scanf("%d%d", &x, &y), add(x, y), add(y, x);
        for (i = 1; i <= n; i++) b[i] = INF;
        b[s] = 0;
        int l = 0, r = 1; q[1] = s;
        while (l < r) {
            int k = q[++l];
            for (int i = head[k]; i; i = nx[i]) if (b[to[i]] == INF) b[to[i]] = b[k] + 1, q[++r] = to[i];
        }
        bb = 0;
        for (i = 1; i <= k; i++) if (b[a[i]] <= i) {
            bb = 1; printf("%d\n", i); break;
        }
        if (!bb) printf("%d\n", b[a[k]]);
    }
    return 0;
}