HDU 2112 HDU Today

最新推荐文章于 2020-07-24 14:10:52 发布

youthinkwu

最新推荐文章于 2020-07-24 14:10:52 发布

阅读量192

点赞数

分类专栏：最短路文章标签： acm 最短路

本文链接：https://blog.csdn.net/youthinkwu/article/details/43890221

版权

最短路专栏收录该内容

3 篇文章 0 订阅

订阅专栏

单源最短路。spfa。这里给出的地名都是字符串。用map映射一下。

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <vector>
#include <queue>
#include <map>
#include <string>
#include <iostream>
using namespace std;
#define N 111111
struct node {
    int v, w;
    node() {}
    node(int _v, int _w):v(_v), w(_w) {}
};
map <string,int> m;
vector<node> g[N];
int n,d[N];
bool inq[N];

int spfa(int s, int t) {
    memset(inq, false, sizeof(inq));
    memset(d, 0x3f, sizeof(d));
    d[s] = 0;
    queue<int> Q;
    Q.push(s);
    while (!Q.empty()) {
        int now = Q.front(); Q.pop();
        inq[now] = false;
        for (int i=0, v, w; i<g[now].size(); i++) {
            v = g[now][i].v, w = g[now][i].w;
            if (d[v] > d[now] + w) {
                d[v] = d[now] + w;
                if (!inq[v]) {
                    inq[v] = true; Q.push(v);
                }
            }
        }
    }
    if(d[t] == 0x3f3f3f3f) return -1;
    return d[t];
}
int main()
{
    while (scanf("%d",&n) == 1)
    {
        if(n == -1) return 0;
        for (int i=0; i<=n; i++) g[i].clear();
        m.clear();
        int num = 0;
        string s,t;
        cin>>s>>t;
        if(!m[s]) m[s] = num++;
        if(!m[t]) m[t] = num++;
        for (int i=0;i<n; i++)
        {
            string s1,s2;
            int w;
            cin>>s1>>s2>>w;
            if(!m[s1]) m[s1] = num++;
            if(!m[s2]) m[s2] = num++;
            g[m[s1]].push_back(node(m[s2], w));
            g[m[s2]].push_back(node(m[s1], w));
        }
        printf("%d\n", spfa(m[s], m[t]));
    }
    return 0;
}