Codeforces Round #368 (Div. 2) B. Bakery （水题）

链接

B. Bakery

---

题意

给出一个n节点m条边的无向图，n个节点中有k个代表面粉厂，你只能将你的面包店开设在剩下n-k个节点上。每条边有一个长度，问把面包店开设在哪个节点上能够离面粉厂距离最近。输出最小距离，无解输出-1。

---

思路

题目很简单，找到离任意一个面粉厂最近的普通节点就可以。
把节点分为两类（放入两个集合或者用数组标记均可），由于输入给出的边的形式是(a, b, l)，我们只要判断a和b是否属于不同集合，如果是，就根据l不断更新答案即可，没什么难度。

---

代码

#include <cstdio>
#include <set>
#include <iostream>
using namespace std;
#define maxn (100010)
struct _node
{
    int u, v, l;
} con[maxn];
int main()
{
    int n, m, k;
    cin >> n >> m >> k;
    for(int i = 0; i < m; i++)
        scanf("%d%d%d", &con[i].u, &con[i].v, &con[i].l);
    set<int> Set;
    int p;
    while(k--)
    {
        scanf("%d", &p);
        Set.insert(p);
    }
    int o = 1000000007;
    for(int i = 0, k1, k2; i < m; i++)
    {
        k1 = k2 = 0;
        if(Set.count(con[i].u)) k1 = 1;
        if(Set.count(con[i].v)) k2 = 1;
        if(k1 + k2 == 1) o = min(o, con[i].l);
    }
    if(o == 1000000007) cout << -1 << endl;
    else cout << o << endl;

    return 0;
}