USACO - Chapter2 Section 2.4 - Bessie Come Home

Bessie Come Home

题目描述

现在是晚餐时间,而母牛们在外面分散的牧场中。 农民约翰按响了电铃,所以她们开始向谷仓走去。 你的工作是要指出哪只母牛会最先到达谷仓(在给出的测试数据中,总会有且只有一只最快的母牛)。 在挤奶的时候(晚餐前),每只母牛都在她自己的牧场上,一些牧场上可能没有母牛。 每个牧场由一条条道路和一个或多个牧场连接(可能包括自己)。 有时，两个牧场(可能是字母相同的)之间会有超过一条道路相连。 至少有一个牧场和谷仓之间有道路连接。 因此,所有的母牛最后都能到达谷仓,并且母牛总是走最短的路径。 当然,母牛能向着任意一方向前进,并且她们以相同的速度前进。 牧场被标记为’a’..’z’和’A’..’Y’,在用大写字母表示的牧场中有一只母牛,小写字母中则没有。 谷仓的标记是’Z’,注意没有母牛在谷仓中。

注意’m’和’M’不是同一个牧场 否则错误 上面的意思是说：输入数据中可能会同时存在M,m（郁闷ing)(PS:表郁闷…告诉我set of咋用就不郁闷了…)，比如

M a a m m z

输入输出格式

输入格式：
第 1 行: 整数 P(1<= P<=10000),表示连接牧场(谷仓)的道路的数目。

第 2 ..P+1行: 用空格分开的两个字母和一个整数:

被道路连接牧场的标记和道路的长度(1<=长度<=1000)。

输出格式：
单独的一行包含二个项目: 最先到达谷仓的母牛所在的牧场的标记,和这只母牛走过的路径的长度。

输入输出样例

输入样例#1：
5
A d 6
B d 3
C e 9
d Z 8
e Z 3
输出样例#1：
B 11

简单的单元最短路，从’Z’到其他点的路，注意只能是大写字母才有牛。

Code

#include<iostream>
#include<vector>
#include<cstdio>
#include<cstring>
#include<queue>
#include<cmath>
using namespace std;

struct edge
{
    int vertex,weight;
    edge(){}
    edge(int vv,int ww){vertex = vv; weight = ww;}
};

vector<vector<edge> > edges(256);

bool in_queue[256];
int dist[256],n,min_dist=0x3f3f3f3f,pos;

void init()
{
    memset(dist,0x3f,sizeof(dist));
    memset(in_queue,0,sizeof(in_queue));
}

void spfa(int stn)
{
    queue<int> q;
    q.push(stn);
    in_queue[stn] = true;
    dist[stn] = 0;
    while(!q.empty())
    {
        int temp = q.front();
        q.pop();
        in_queue[temp] = false;
        for (int i=0;i<edges[temp].size();i++)
        {
            int vtx = edges[temp][i].vertex;
            if (dist[temp] + edges[temp][i].weight < dist[vtx])
            {
                dist[vtx] = dist[temp] + edges[temp][i].weight;
                if (!in_queue[vtx])
                {
                    q.push(vtx);
                    in_queue[vtx] = true;
                }
            }
        }
    }
}

int main()
{
    init();
    cin >> n;
    for (int i=0;i<n;i++)
    {
        char fm,to;
        int wt;
        cin >> fm >> to >> wt;
        edges[int(to)].push_back(edge(int(fm),wt));
        edges[int(fm)].push_back(edge(int(to),wt));
    }
    spfa(int('Z'));
    for (int i=65;i<=89;i++)
        if (min_dist > dist[i])
        {
            min_dist = dist[i];
            pos = i;
        }
    cout << char(pos) << " " << min_dist;
    return 0;
}