CodeForces 707B-Bakery

题目链接

题目分析

题意

有n个城市，其中k个设有仓库，城市中有m条路连通。每两个城市u和v间道路长度为l千米，每千米需花费1卢布，求能满足从设有仓库的城市到另一个能连通的城市的最少花费。没有满足条件的则输出-1。

思路

仓库k小于等于0时输出-1。若k>0，将每条路按距离从小到大排列。输入可设仓库的城市编号a时，用f[a]将其标记为1，若可连通的城市u、v满足f[u]、f[v]一个标记一个未标记则输出对应的距离l即等于所求。否则输出-1.

代码

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string.h>
using namespace std; 
#define maxn 100001
int f[maxn];
struct node{
    int u,v;
    long long l;
}x[maxn];
bool cmp(node a,node b){
    return a.l<b.l;
}
int main(){
    //freopen("data.in","r",stdin);
    int n,m,k;
    while(cin>>n>>m>>k){
        memset(f,0,sizeof(f));
        int i,a;
        if(k<=0||k==n){
            cout<<"-1"<<endl;
            break;
        }   
        for(i=0;i<m;i++)
            cin>>x[i].u>>x[i].v>>x[i].l;
        for(i=0;i<k;i++){
            cin>>a;
            f[a]=1;
        }
        sort(x,x+m,cmp);
        for(i=0;i<m;i++){
             if(f[x[i].u]==0&&f[x[i].v]==1||f[x[i].u]==1&&f[x[i].v]==0){
                cout<<x[i].l<<endl;
                break;
             }
        }
        if(i==m)
            cout<<"-1"<<endl;
    }
    return 0;
}