喊山（数据结构习题）

喊山，是人双手围在嘴边成喇叭状，对着远方高山发出“喂—喂喂—喂喂喂……”的呼唤。呼唤声通过空气的传递，回荡于深谷之间，传送到人们耳中，发出约定俗成的“讯号”，达到声讯传递交流的目的。原来它是彝族先民用来求援呼救的“讯号”，慢慢地人们在生活实践中发现了它的实用价值，便把它作为一种交流工具世代传袭使用。

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

输入格式：

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

输出格式：

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

#include <queue>
#include <iostream>
#include <vector>
#include <set>
using namespace std;

void fill_vv(vector<vector<int>>& vv, int m)
{
    while (m != 0)
    {
        // 输入两座山的号码
        int m1 = 0;
        int m2 = 0;
        scanf("%d", &m1);
        scanf("%d", &m2);
        vv[m1].push_back(m2);
        vv[m2].push_back(m1);
        m--;
    }
}

void fill_inquire(vector<int>& inquire, int k)
{
    while (k != 0)
    {
        // 接收需要查询的山头号码
        int x = 0;
        scanf("%d", &x);
        inquire.push_back(x);
        k--;
    }
}

// 负责寻找山头号码为k的山的声音,所能传达到最远的山头
int find_farthest(vector<vector<int>>& vv, int k)
{
    queue<int> q;
    vector<vector<int>> distance_vv(vv.size());
    set<int> baned_num;
    // 一开始队列为空，把需要查询的山头入队
    q.push(k);
    // 每入一个数据的同时，讲该数据ban掉
    baned_num.insert(k);
    int distance = 0;
    while (!q.empty())
    {
        int q_size = q.size();
        while (q_size != 0)
        {
            int tmp = q.front();
            distance_vv[distance].push_back(tmp);
            int size = vv[tmp].size();
            q.pop();
            //int flag = 0;
            for (int i = 0; i < size; i++)
            {
                if (baned_num.count(vv[tmp][i]) == 0)
                {
                    baned_num.insert(vv[tmp][i]);
                    q.push(vv[tmp][i]);
                    //flag = 1;
                }
            }
            q_size--;
        }
        distance++;
        
    }

    distance--;
    if (distance <= 0)
    {
        return 0;
    }
    
    if (distance_vv[distance].size() == 1)
    {
        return distance_vv[distance][0];
    }
    else
    {
        return (distance_vv[distance][0] < distance_vv[distance][1] ? distance_vv[distance][0] : distance_vv[distance][1]);
    }
}

void solution(vector<vector<int>>& vv, vector<int>& inquire, vector<int>& answer)
{
    // inquire数组的大小
    int size = inquire.size();
    for (int i = 0; i < size; i++)
    {
        int ret = find_farthest(vv, inquire[i]);
        answer.push_back(ret);
    }
}

void print_vector(vector<int> v)
{
    for (auto& e : v)
    {
        cout << e << endl;
    }
}

int main()
{
    int n = 0; // 山的总数
    int m = 0; // 关系个数
    int k = 0; // 需要查询的个数
    scanf("%d", &n);
    scanf("%d", &m);
    scanf("%d", &k);
    vector<vector<int>> vv(n + 1);
    fill_vv(vv, m);
    vector<int> inquire;
    fill_inquire(inquire, k);
    vector<int> answer;
    solution(vv, inquire, answer);
    print_vector(answer);
    return 0;
}