PAT 天梯赛 L3-008. 喊山 【BFS】

题目链接

https://www.patest.cn/contests/gplt/L3-008

思路
因为 每个山头 最多有两个 能听到它的 临近山头
那么 我们就可以 给每个 山头 都 分配 最多两个子儿子
然后 再从儿子 往下找
一层一层往下找 如果找的元素 之前已经出现过 这个 就不压入 队列
直到最后 队列为空 输出当层 编号最小的那个

AC代码

#include <cstdio>
#include <cstring>
#include <ctype.h>
#include <cstdlib>
#include <cmath>
#include <climits>
#include <ctime>
#include <iostream>
#include <algorithm>
#include <deque>
#include <vector>
#include <queue>
#include <string>
#include <map>
#include <stack>
#include <set>
#include <numeric>
#include <sstream>
#include <iomanip>
#include <limits>

#define CLR(a) memset(a, 0, sizeof(a))

using namespace std;
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
typedef pair <int, int> pii;
typedef pair <ll, ll> pll;
typedef pair<string, int> psi;
typedef pair<string, string> pss;

const double PI = 3.14159265358979323846264338327;
const double E = exp(1);
const double eps = 1e-3;

const int INF = 0x3f3f3f3f;
const int maxn = 1e4 + 5;
const int MOD = 1e9 + 7;

int ans;

queue <int> opt;

map <int, int> q, vis;

struct Node
{
    int l, r;
}w[maxn];

void bfs(int cur)
{
    int len = opt.size();
    int Min = INF;
    while (len--)
    {
        int num = opt.front();
        opt.pop();
        if (num < Min)
            Min = num;
        if (vis[w[num].l] == 0)
        {
            opt.push(w[num].l);
            vis[w[num].l] = 1;
        }
        if (vis[w[num].r] == 0)
        {
            opt.push(w[num].r);
            vis[w[num].r] = 1;
        }
    }
    if (opt.empty())
    {
        ans = Min;
        return;
    }
    else
        bfs(cur + 1);
}

int main()
{
    int n, m, k;
    scanf("%d%d%d", &n, &m, &k);
    int a, b;
    for (int i = 0; i < m; i++)
    {
        scanf("%d%d", &a, &b);
        q[a] = 1;
        q[b] = 1;
        if (w[a].l)
            w[a].r = b;
        else
            w[a].l = b;
        if (w[b].l)
            w[b].r = a;
        else
            w[b].l = a;
    }
    for (int i = 0; i < k; i++)
    {
        scanf("%d", &a);
        if (q[a] == 0)
            printf("0
");
        else
        {
            vis.clear();
            vis[0] = 1;
            while (!opt.empty())
                opt.pop();
            if (w[a].l != 0)
                opt.push(w[a].l);
            if (w[a].r != 0)
                opt.push(w[a].r);
            vis[w[a].l] = 1;
            vis[w[a].r] = 1;
            vis[a] = 1;
            bfs(1);
            printf("%d
", ans);
        }
    }
}